Update name of model
Browse files
inference_agent_emotion_classifier.py
CHANGED
|
@@ -78,7 +78,7 @@ def _resolve_paths() -> Tuple[str, str]:
|
|
| 78 |
)
|
| 79 |
|
| 80 |
print("⬇️ Descargando archivos desde Hugging Face Hub...")
|
| 81 |
-
ckpt_path = hf_hub_download(repo_id=REPO_ID, filename="
|
| 82 |
cfg_path = hf_hub_download(repo_id=REPO_ID, filename="config_agent.json")
|
| 83 |
return ckpt_path, cfg_path
|
| 84 |
|
|
|
|
| 78 |
)
|
| 79 |
|
| 80 |
print("⬇️ Descargando archivos desde Hugging Face Hub...")
|
| 81 |
+
ckpt_path = hf_hub_download(repo_id=REPO_ID, filename="best_model_agent.pt")
|
| 82 |
cfg_path = hf_hub_download(repo_id=REPO_ID, filename="config_agent.json")
|
| 83 |
return ckpt_path, cfg_path
|
| 84 |
|