Swap out for a more performant model

app.py

@@ -3,12 +3,11 @@ from transformers import pipeline
 import os
 import logging
 
-
 # Set Hugging Face cache directory
 os.environ["HF_HOME"] = "/app/.cache"
 
 app = FastAPI()
-classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+classifier = pipeline("zero-shot-classification", model="valhalla/distilbart-mnli-12-1")
 
 @app.get("/")
 async def root():
@@ -22,4 +21,3 @@ async def predict(data: dict):
     result = classifier(text, labels)
     logging.info(f"Prediction result: {result}")
     return {"label": result["labels"][0]}
-