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 = "HuggingFaceH4/zephyr-7b-beta"
-fallbacks = ["mistralai/Mistral-7B-Instruct-v0.2", "HuggingFaceH4/zephyr-7b-beta"]
+primary = "mistralai/Mistral-7B-Instruct-v0.2"
+fallbacks = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mistral-7B-Instruct-v0.2"]
 
 # Load Hugging Face API Key
 HF_API_KEY = os.getenv("hf_...XHjt")  # Get API key from environment variables