Update main.py

main.py

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