Update app.py

app.py

@@ -1,8 +1,19 @@
+import os
 import gradio as gr
 from transformers import pipeline
+from huggingface_hub import login
 
-# Load Mistral-7B-Instruct AI model
-chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
+# Get Hugging Face token securely from environment variables
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# Authenticate with Hugging Face
+if HF_TOKEN:
+    login(HF_TOKEN)
+else:
+    raise ValueError("🚨 Hugging Face token not found! Please add it to Secrets in your Hugging Face Space.")
+
+# Load the Mistral-7B-Instruct AI model
+chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1", token=HF_TOKEN)
 
 def generate_itinerary(destination, start_date, end_date, traveler_type, companion):
     """Generates an AI-powered travel itinerary based on user preferences"""