Update app.py

app.py

@@ -13,10 +13,8 @@ model.config.pad_token_id = model.config.eos_token_id
 
 # Chat function
 def chat_with_gpt2(user_input):
-    # Encode input
     inputs = tokenizer.encode(user_input, return_tensors="pt", truncation=True, max_length=512)
     
-    # Generate response (limit output for speed and memory)
     outputs = model.generate(
         inputs,
         max_length=250,  # shortened for performance
@@ -28,14 +26,16 @@ def chat_with_gpt2(user_input):
         pad_token_id=tokenizer.eos_token_id
     )
     
-    # Decode response
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    
-    # Return only the bot's reply (not repeating input)
     return response[len(user_input):].strip()
 
 # Gradio UI
 interface = gr.Interface(
     fn=chat_with_gpt2,
     inputs=gr.Textbox(lines=3, placeholder="Ask me about your trip..."),
-    outputs="
+    outputs="text",
+    title="Trip Planner Chatbot",
+    description="Ask anything related to your travel or itinerary!"
+)
+
+interface.launch()