Update main.py

main.py

@@ -7,8 +7,8 @@ import uvicorn  # Import uvicorn for running the FastAPI application
 app = FastAPI()  # Create a FastAPI instance
 
 # Define the primary and fallback models
-primary = "mistralai/Mistral-7B-Instruct-v0.2"
-fallbacks = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mistral-7B-Instruct-v0.2"]
+primary = "Qwen2.5-0.5B"
+fallbacks = ["mistralai/Mistral-7B-Instruct-v0.2", "facebook/bart-large-mnli"]
 
 # Load Hugging Face API Key
 HF_API_KEY = os.getenv("HF_API_TOKEN")  # Get API key from environment variables