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
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -19,7 +19,7 @@ from config import (
19
  API_DESCRIPTION,
20
  API_VERSION,
21
  STANCE_MODEL_ID,
22
- HUGGINGFACE_STANCE_API_KEY,
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, HUGGINGFACE_STANCE_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!")
 
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!")