Update app.py

app.py

@@ -12,22 +12,22 @@ if HF_TOKEN:
 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="mistralai/Mistral-7B-8Bit", token=HF_TOKEN)
+# Use Phi-2 (optimized for free-tier)
+chatbot = pipeline("text-generation", model="microsoft/phi-2", token=HF_TOKEN)
 
 def generate_itinerary(destination, start_date, end_date, traveler_type, companion):
     """Generates an AI-powered travel itinerary based on user preferences"""
     
     prompt = f"""
-    You are a travel assistant. Generate a detailed day-by-day itinerary for a trip to {destination}.
-    - Trip Duration: {start_date} to {end_date}
-    - Type of traveler: {traveler_type}
-    - Traveling with: {companion}
-    - Each day should have 3 activities for morning, afternoon, and night.
-    - Provide unique local recommendations based on the traveler's interests.
+    You are an AI travel assistant. Generate a detailed itinerary for {destination}.
+    - Dates: {start_date} to {end_date}
+    - Traveler type: {traveler_type}
+    - Companion: {companion}
+    - Provide 3 activities per day (Morning, Afternoon, Night)
+    - Keep responses short and optimized for fast generation.
     """
     
-    response = chatbot(prompt, max_length=150, do_sample=True, temperature=0.7, top_p=0.9)
+    response = chatbot(prompt, max_length=200, do_sample=True, temperature=0.7, top_p=0.9)
     return response[0]['generated_text']
 
 # Creating a simple user interface