Update app.py

app.py

@@ -12,12 +12,12 @@ if HF_TOKEN:
 else:
     raise ValueError("🚨 Hugging Face token not found! Please add it to Secrets in your Hugging Face Space.")
 
-# Load AI model (optimized for T4 Small GPU)
+# Load AI model (Optimized for Hugging Face GPU)
 chatbot = pipeline(
     "text-generation",
-    model="mistralai/Mistral-7B-Instruct-v0.1",
+    model="TheBloke/Mistral-7B-Instruct-v0.1-GGUF",  # Updated to GGUF version
     token=HF_TOKEN,
-    device=0  # Runs on GPU
+    device=0  # Ensures it runs on GPU
 )
 
 def generate_itinerary(destination, start_date, end_date, traveler_type, companion):
@@ -39,7 +39,7 @@ def generate_itinerary(destination, start_date, end_date, traveler_type, compani
     - 🌙 **Night**: Experience [Cultural/Nightlife Activity]
     """
 
-    response = chatbot(prompt, max_length=300, do_sample=True, temperature=0.6, top_p=0.85)
+    response = chatbot(prompt, max_length=250, do_sample=True, temperature=0.6, top_p=0.85)
     return response[0]['generated_text']
 
 # Creating a simple user interface