Update app.py

app.py

@@ -12,22 +12,34 @@ if HF_TOKEN:
 else:
     raise ValueError("🚨 Hugging Face token not found! Please add it to Secrets in your Hugging Face Space.")
 
-# Use Phi-2 (optimized for free-tier)
-chatbot = pipeline("text-generation", model="microsoft/phi-2", token=HF_TOKEN)
+# Load AI model (optimized for T4 Small GPU)
+chatbot = pipeline(
+    "text-generation",
+    model="mistralai/Mistral-7B-Instruct-v0.1",
+    token=HF_TOKEN,
+    device=0  # Runs on GPU
+)
 
 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 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.
+    You are a professional travel assistant. Create a **day-by-day itinerary** for a trip to {destination}.
+    - 📅 **Dates**: {start_date} to {end_date}
+    - 🎭 **Traveler Type**: {traveler_type}
+    - 🧳 **Companion**: {companion}
+    - Provide **3 activities per day**: 🌅 Morning, ☀️ Afternoon, 🌙 Night.
+    - Include **real attractions, famous landmarks, hidden gems, and local experiences**.
+    - Keep responses **structured, short, and engaging**.
+
+    📌 **Example Format**:
+    **Day 1**:
+    - 🌅 **Morning**: Visit [Famous Landmark] in {destination}
+    - ☀️ **Afternoon**: Try [Local Food] at [Best Restaurant]
+    - 🌙 **Night**: Experience [Cultural/Nightlife Activity]
     """
-    
-    response = chatbot(prompt, max_length=200, do_sample=True, temperature=0.7, top_p=0.9)
+
+    response = chatbot(prompt, max_length=300, do_sample=True, temperature=0.6, top_p=0.85)
     return response[0]['generated_text']
 
 # Creating a simple user interface