Update app.py

app.py

@@ -165,7 +165,7 @@ def main():
         from streamlit_chat import message as st_message
         from transformers import BlenderbotTokenizer
         from transformers import BlenderbotForConditionalGeneration
-        st.title("Simple Chatbot for fun!")
+        st.title("Chatbot!!!")
         
         @st.experimental_singleton
         def get_models():
@@ -177,22 +177,21 @@ def main():
             return tokenizer, model
         if "history" not in st.session_state:
             st.session_state.history = []
-        st.title("Hello Chatbot")
-        def main():
-            st.text_input("Talk to the bot", key="input_text", on_change=generate_answer)
-            def generate_answer():
-                tokenizer, model = get_models()
-                user_message = st.session_state.input_text
-                inputs = tokenizer(st.session_state.input_text, return_tensors="pt")
-                result = model.generate(**inputs)
-                message_bot = tokenizer.decode(
-                    result[0], skip_special_tokens=True
-                )  # .replace("<s>", "").replace("</s>", "")
-                st.session_state.history.append({"message": user_message, "is_user": True})
-                st.session_state.history.append({"message": message_bot, "is_user": False})
-            from copyreg import clear_extension_cache
-            for chat in st.session_state.history:
-                st_message(**chat)
+        st.title("Hello bot: ")
+        st.text_input("Talk to the bot", key="input_text", on_change=generate_answer)
+        def generate_answer():
+            tokenizer, model = get_models()
+            user_message = st.session_state.input_text
+            inputs = tokenizer(st.session_state.input_text, return_tensors="pt")
+            result = model.generate(**inputs)
+            message_bot = tokenizer.decode(
+                result[0], skip_special_tokens=True
+            )  # .replace("<s>", "").replace("</s>", "")
+            st.session_state.history.append({"message": user_message, "is_user": True})
+            st.session_state.history.append({"message": message_bot, "is_user": False})
+        from copyreg import clear_extension_cache
+        for chat in st.session_state.history:
+            st_message(**chat)
 
 
 if __name__ == "__main__":