Update app.py

app.py

@@ -17,7 +17,7 @@ import os
 app = FastAPI()
 
 print(f"🔄 Loading models")
-model_name = "mistralai/Mistral-7B-v0.1"
+model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")  # Fetch token from environment variable
 
 tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)