Update app.py

app.py

@@ -1,11 +1,14 @@
 from fastapi import FastAPI
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, AutoConfig
 import torch
 
+# Verify model configuration
+config = AutoConfig.from_pretrained("SrivarshiniGanesan/finetuned-stress-model")
+print(config)
 # Initialize FastAPI app
 app = FastAPI()
 
-model = AutoModelForSequenceClassification.from_pretrained("SrivarshiniGanesan/finetuned-stress-model")
+model = AutoModelForSequenceClassification.from_pretrained("SrivarshiniGanesan/finetuned-stress-model",config=config)
 tokenizer = AutoTokenizer.from_pretrained("SrivarshiniGanesan/finetuned-stress-model")
 
 @app.get("/")