Yassine Mhirsi
commited on
Commit
·
dc6c29a
1
Parent(s):
70a2026
Fix variable name for Hugging Face API key in stance model loading process in main.py
Browse files
main.py
CHANGED
|
@@ -19,7 +19,7 @@ from config import (
|
|
| 19 |
API_DESCRIPTION,
|
| 20 |
API_VERSION,
|
| 21 |
STANCE_MODEL_ID,
|
| 22 |
-
|
| 23 |
HUGGINGFACE_STANCE_MODEL_ID,
|
| 24 |
HOST,
|
| 25 |
PORT,
|
|
@@ -46,7 +46,7 @@ async def lifespan(app: FastAPI):
|
|
| 46 |
# Load stance detection model
|
| 47 |
try:
|
| 48 |
logger.info(f"Loading stance model from Hugging Face: {HUGGINGFACE_STANCE_MODEL_ID}")
|
| 49 |
-
stance_model_manager.load_model(HUGGINGFACE_STANCE_MODEL_ID,
|
| 50 |
except Exception as e:
|
| 51 |
logger.error(f"✗ Failed to load stance model: {str(e)}")
|
| 52 |
logger.error("⚠️ Stance detection endpoints will not work!")
|
|
|
|
| 19 |
API_DESCRIPTION,
|
| 20 |
API_VERSION,
|
| 21 |
STANCE_MODEL_ID,
|
| 22 |
+
HUGGINGFACE_API_KEY,
|
| 23 |
HUGGINGFACE_STANCE_MODEL_ID,
|
| 24 |
HOST,
|
| 25 |
PORT,
|
|
|
|
| 46 |
# Load stance detection model
|
| 47 |
try:
|
| 48 |
logger.info(f"Loading stance model from Hugging Face: {HUGGINGFACE_STANCE_MODEL_ID}")
|
| 49 |
+
stance_model_manager.load_model(HUGGINGFACE_STANCE_MODEL_ID, HUGGINGFACE_API_KEY)
|
| 50 |
except Exception as e:
|
| 51 |
logger.error(f"✗ Failed to load stance model: {str(e)}")
|
| 52 |
logger.error("⚠️ Stance detection endpoints will not work!")
|