updating the app

app.py

@@ -131,14 +131,14 @@ def main():
     #     except KeyError:
     #         st.error(f"Unexpected response format. Missing 'response_text' key. Received: {response.text}")
 
-    if prompt := st.chat_input("What is up?"):
-        response = pipe(f"<s>[INST] {prompt} [/INST]")
-        response_text = response[0]["generated_text"].replace("<s>[INST]", "").replace("[/INST]", "").strip()
-
-    with st.chat_message("assistant"):
-        st.markdown(response_text)
-
-    st.session_state.messages.append({"role": "assistant", "content": response_text})
+        if prompt := st.chat_input("What is up?"):
+            response = pipe(f"<s>[INST] {prompt} [/INST]")
+            response_text = response[0]["generated_text"].replace("<s>[INST]", "").replace("[/INST]", "").strip()
+    
+        with st.chat_message("assistant"):
+            st.markdown(response_text)
+    
+        st.session_state.messages.append({"role": "assistant", "content": response_text})
 
         # Save updated chat history
         user_data["chat_history"] = st.session_state.messages