Update app.py

app.py

@@ -13,7 +13,7 @@ else:
     raise ValueError("🚨 Hugging Face token not found! Please add it to Secrets in your Hugging Face Space.")
 
 # Load a smaller AI model that fits within 16GB memory
-chatbot = pipeline("text-generation", model="microsoft/phi-2", token=HF_TOKEN)
+chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-8Bit", token=HF_TOKEN)
 
 def generate_itinerary(destination, start_date, end_date, traveler_type, companion):
     """Generates an AI-powered travel itinerary based on user preferences"""
@@ -27,7 +27,7 @@ def generate_itinerary(destination, start_date, end_date, traveler_type, compani
     - Provide unique local recommendations based on the traveler's interests.
     """
     
-    response = chatbot(prompt, max_length=500, do_sample=True)
+    response = chatbot(prompt, max_length=150, do_sample=True, temperature=0.7, top_p=0.9)
     return response[0]['generated_text']
 
 # Creating a simple user interface