changes to app.py

app.py

@@ -62,4 +62,7 @@ def main():
         st.chat_message("user").markdown(prompt)
         model_response = respond_to_question(prompt, model, tokenizer)
         st.session_state.messages.append({'role': 'Assistant', 'text': model_response})
-        st.chat_message("system").markdown(model_response)
+        st.chat_message("system").markdown(model_response)
+        
+if __name__ == "__main__":
+    main()