Update main.py

main.py

@@ -7,8 +7,8 @@ import uvicorn
 app = FastAPI()  # Create FastAPI instance
 
 # Define the primary and fallback models
-primary = "mistralai/Mixtral-8x7B-Instruct-v0.1"
-fallbacks = ["mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mixtral-8x7B-Instruct-v0.1"]
+primary = "meta-llama/Llama-3.2-3B-Instruct"
+fallbacks = ["meta-llama/Llama-3.2-3B-Instruct", "meta-llama/Llama-3.2-3B-Instruct"]
 
 # Load Hugging Face API Key
 HF_API_KEY = os.getenv("HF_API_TOKEN")  # Get API key from environment variables