Update server.py

server.py

@@ -7,17 +7,24 @@ from transformers import pipeline, AutoModelForSequenceClassification, AutoToken
 import os
 
 app = FastAPI()
-os.environ["huggingfacetoken"] = "/app/.cache"
-model_name = "facebook/bart-large-mnli"
+# os.environ["huggingfacetoken"] = "/app/.cache"
+# model_name = "facebook/bart-large-mnli"
 
 # Force PyTorch model instead of Flax
-model = AutoModelForSequenceClassification.from_pretrained(model_name, from_flax=True)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+# model = AutoModelForSequenceClassification.from_pretrained(model_name, from_flax=True)
+# tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+model = AutoModelForSequenceClassification.from_pretrained(
+    "facebook/bart-large-mnli",
+    force_download=True  # Forces re-download
 
 classifier = pipeline("zero-shot-classification", model=model, tokenizer=tokenizer)
 
 categories = ["Spam", "Not Spam"]
 
+)
+
+
 class EmailCredentials(BaseModel):
     email: str
     password: str