Update app.py

app.py

@@ -4,17 +4,27 @@ from huggingface_hub import InferenceClient
 # Initialize the client with your desired model
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# Define the system prompt as an AI Dermatologist
+# # Define the system prompt as an AI Dermatologist
+# def format_prompt(message, history):
+#     prompt = "<s>"
+#     # Start the conversation with a system message
+#     prompt += "[INST] You are a Travel Companion Chatbot that helps users plan trips by suggesting transport, sightseeing stops, and accommodations based on their preferences. [/INST]"
+#     for user_prompt, bot_response in history:
+#         prompt += f"[INST] {user_prompt} [/INST]"
+#         prompt += f" {bot_response}</s> "
+#     prompt += f"[INST] {message} [/INST]"
+#     return prompt
 def format_prompt(message, history):
     prompt = "<s>"
     # Start the conversation with a system message
-    prompt += "[INST] You are a Travel Companion Chatbot that helps users plan trips by suggesting transport, sightseeing stops, and accommodations based on their preferences. [/INST]"
-    for user_prompt, bot_response in history:
-        prompt += f"[INST] {user_prompt} [/INST]"
-        prompt += f" {bot_response}</s> "
+    prompt += "[INST] You are a Travel Companion Chatbot that helps users plan trips by suggesting transport, sightseeing stops, and accommodations based on their preferences. Please assist the user by asking what they need to know. [/INST]"
+    
+    # Only append the user message, without the historical responses or examples
     prompt += f"[INST] {message} [/INST]"
+    
     return prompt
 
+
 # Function to generate responses with the AI Dermatologist context
 def generate(
     prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0