Update app.py

app.py

@@ -66,7 +66,7 @@ class MistralChatBot:
 
         sampling_params = SamplingParams(temperature=0.7, top_p=.95, max_tokens=500, presence_penalty = 2)
         detailed_prompt = "<|im_start|>system\n" + system_prompt + "<|im_end|>\n<|im_start|>user"""
-        detailed_prompt = detailed_prompt + "\n" + user_input + "<|im_end|>\n<|im_start|>assistant\n"
+        detailed_prompt = detailed_prompt + "\n" + user_message + "<|im_end|>\n<|im_start|>assistant\n"
         prompts = [detailed_prompt]
         outputs = llm.generate(prompts, sampling_params, use_tqdm = False)
         generated_text = outputs[0].outputs[0].text
@@ -77,7 +77,7 @@ class MistralChatBot:
         conversation = "<|im_start|>system\n" + system_prompt + "<|im_end|>\n<|im_start|>user\n" + user_message + "<|im_end|>\n<|im_start|>assistant\n"
         return conversation
 
-# Create the Falcon chatbot instance
+# Create the Mistral chatbot instance
 mistral_bot = MistralChatBot()
 
 # Define the Gradio interface