Update app.py

app.py

@@ -83,6 +83,11 @@ def reset_conversation():
     st.session_state.messages = []
     return None
     
+def format_promt(message, custom_instructions=None):
+    if custom_instructions:
+        promt += f"[INST] {custom_instructions} [/INST]"
+    promt += f"[INST] {message} [/INST]"
+    return promt
 
 
 
@@ -146,6 +151,10 @@ for message in st.session_state.messages:
 # Accept user input
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
 
+    custom_instruction = "Act like a Human in conversation"
+
+    formated_text = format_promt(prompt, custom_instruction)
+
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
@@ -160,10 +169,7 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
             headers=headers)
 
         output = client.text_generation(
-            Prompt=[
-                {"role": m["role"], "content": m["content"]}
-                for m in st.session_state.messages
-            ],
+            formated_text
             temperature=temp_values,#0.5
             max_tokens=3000,
         )