malek-messaoudii commited on
Commit
1438c6f
·
1 Parent(s): 3afc11b

Update training script and model files

Browse files
Files changed (2) hide show
  1. config.py +1 -1
  2. main.py +0 -1
config.py CHANGED
@@ -18,7 +18,7 @@ HUGGINGFACE_LABEL_MODEL_ID = os.getenv("HUGGINGFACE_LABEL_MODEL_ID")
18
 
19
  # Use Hugging Face model ID instead of local path
20
  STANCE_MODEL_ID = HUGGINGFACE_STANCE_MODEL_ID
21
- LABEL_MODEL_ID = HUGGINGFACE_LABEL_MODEL_ID
22
 
23
  # API configuration
24
  API_TITLE = "NLP Project API"
 
18
 
19
  # Use Hugging Face model ID instead of local path
20
  STANCE_MODEL_ID = HUGGINGFACE_STANCE_MODEL_ID
21
+ STANCE_LABEL_ID = HUGGINGFACE_LABEL_MODEL_ID
22
 
23
  # API configuration
24
  API_TITLE = "NLP Project API"
main.py CHANGED
@@ -69,7 +69,6 @@ async def lifespan(app: FastAPI):
69
  # Shutdown: Cleanup (if needed)
70
  # Currently no cleanup needed, but you can add it here if necessary
71
 
72
-
73
  # Create FastAPI application
74
  app = FastAPI(
75
  title=API_TITLE,
 
69
  # Shutdown: Cleanup (if needed)
70
  # Currently no cleanup needed, but you can add it here if necessary
71
 
 
72
  # Create FastAPI application
73
  app = FastAPI(
74
  title=API_TITLE,