Update engine/parser_llm.py
Browse files- engine/parser_llm.py +1 -1
engine/parser_llm.py
CHANGED
|
@@ -31,7 +31,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
| 31 |
# ✅ Your fine-tuned model (can be overridden via env var)
|
| 32 |
DEFAULT_MODEL = os.getenv(
|
| 33 |
"BACTAI_LLM_PARSER_MODEL",
|
| 34 |
-
"EphAsad/
|
| 35 |
)
|
| 36 |
|
| 37 |
# ✅ Few-shot OFF by default now (fine-tune doesn't need it)
|
|
|
|
| 31 |
# ✅ Your fine-tuned model (can be overridden via env var)
|
| 32 |
DEFAULT_MODEL = os.getenv(
|
| 33 |
"BACTAI_LLM_PARSER_MODEL",
|
| 34 |
+
"EphAsad/BactAID-v2",
|
| 35 |
)
|
| 36 |
|
| 37 |
# ✅ Few-shot OFF by default now (fine-tune doesn't need it)
|