update

app.py

@@ -26,7 +26,7 @@ app = FastAPI()
 # Add CORS middleware to allow frontend requests
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Allow all origins (you can restrict this later)
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -39,8 +39,8 @@ logger.info("Loading the model...")
 try:
     classifier = pipeline(
         "zero-shot-classification",
-        model="asafaya/bert-base-arabic",  # Use Arabic model
-        tokenizer="asafaya/bert-base-arabic",
+        model="UBC-NLP/ARBERTv2",  # Switch to a better Arabic model
+        tokenizer="UBC-NLP/ARBERTv2",
         cache_dir=cache_dir
     )
     logger.info("Model loaded successfully!")