Update app.py

app.py

@@ -106,7 +106,7 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     instructions = "Respond as an expert doctor in diabetes. Ensure the response is informative and directly answers the user's question."
 
     # Create the full prompt with instructions
-    full_prompt = f"<s>[INST] {instructions} [Q] {prompt} [/Q] [/INST]</s>"
+    full_prompt = f"<s>[INST] {user_prompt} [/INST]  {instructions}</s>"
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
@@ -120,8 +120,8 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
             stream=True,
             max_tokens=1024,
         )
-
         response = st.write_stream(stream)
+        # Remove the '</s>' from the response if it exists
+        response = response.replace('</s>', '').strip()
     st.session_state.messages.append(
-        {"role": "assistant", "content": response})
-
+        {"role": "assistant", "content": response})