Update main.py

main.py

@@ -7,7 +7,13 @@ from transformers import pipeline, AutoModelForSequenceClassification, AutoToken
 
 app = FastAPI()
 
-classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli", from_flax=True)
+model_name = "facebook/bart-large-mnli"
+
+# Force PyTorch model instead of Flax
+model = AutoModelForSequenceClassification.from_pretrained(model_name, from_pt=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+classifier = pipeline("zero-shot-classification", model=model, tokenizer=tokenizer)
 
 categories = ["Spam", "Not Spam"]