Update app.py

app.py

@@ -14,8 +14,6 @@ def query(payload):
     return response.json(), response
 
 
-
-
 context = "To extract information from documents, use sentence similarity task. To do sentiment analysis from tweets, use text classification task. To detect masks from images, use object detection task. To extract information from invoices, use named entity recognition from token classification task."
 
 
@@ -24,6 +22,11 @@ message_history = [{"text":"Let's find out the best task for your use case! Tell
 
 for msg in message_history:
     message(msg["text"], is_user = msg["is_user"])   # display all the previous message
+  
+input = st.text_input("Ask me 🤗")
+
+message_history.append({"text":input, "is_user" : True})
+
 
 placeholder = st.empty()  # placeholder for latest message
 
@@ -32,10 +35,7 @@ with placeholder.container():
     if last_message != "":
     
         message(last_message["text"], last_message["is_user"]) # display the latest message
-        
-input = st.text_input("Ask me 🤗")
-
-message_history.append({"text":input, "is_user" : True})
+      
 
 data, resp = query(
     {