vic3610 commited on
Commit
512cdf5
·
verified ·
1 Parent(s): b504bd1

Update analyze_bob_hf.py

Browse files
Files changed (1) hide show
  1. analyze_bob_hf.py +2 -1
analyze_bob_hf.py CHANGED
@@ -23,7 +23,8 @@ except Exception:
23
  BASE_DIR = Path(os.environ.get("BOB_BASE_DIR", Path(__file__).parent.parent))
24
  TRANSCRIPTIONS_DIR = Path(os.environ.get("BOB_TRANSCRIPTIONS_DIR", BASE_DIR / "output" / "transcriptions"))
25
  OUTPUT_FILE = Path(os.environ.get("BOB_OUTPUT_FILE", BASE_DIR / "output" / "resume_bob.txt"))
26
- HF_MODEL = os.environ.get("HF_MODEL", "facebook/opt-350m") # Modèle léger par défaut
 
27
 
28
  def load_hf_model():
29
  """Charge un modèle Hugging Face"""
 
23
  BASE_DIR = Path(os.environ.get("BOB_BASE_DIR", Path(__file__).parent.parent))
24
  TRANSCRIPTIONS_DIR = Path(os.environ.get("BOB_TRANSCRIPTIONS_DIR", BASE_DIR / "output" / "transcriptions"))
25
  OUTPUT_FILE = Path(os.environ.get("BOB_OUTPUT_FILE", BASE_DIR / "output" / "resume_bob.txt"))
26
+ HF_MODEL = os.environ.get("HF_MODEL", "Qwen/Qwen2-7B-Instruct") # Qwen 7B par défaut
27
+
28
 
29
  def load_hf_model():
30
  """Charge un modèle Hugging Face"""