Update analyze_bob_hf.py
Browse files- 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", "
|
|
|
|
| 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"""
|