Update app.py

app.py

@@ -31,7 +31,7 @@ def load_model():
 def chat(inp, history, agent):
     history = history or []
     if agent is None:
-        history.append((inp, "Please click Load Model or wait for model to load"))
+        history.append((inp, "Please wait for model to load (3-5 seconds)"))
         return history, history
     print("\n==== date/time: " + str(datetime.datetime.now()) + " ====")
     print("inp: " + inp)
@@ -49,11 +49,6 @@ with block:
     with gr.Row():
         gr.Markdown("<h3><center>🤗 Transformers Chat</center></h3><p>Ask questions about the Hugging Face Transformers Library, Powered by Flan-UL2</p>")
 
-        load_model_button = gr.Button(
-            value="Load Model",
-            variant="secondary"
-        ).style(full_width=False)
-
     chatbot = gr.Chatbot()
 
     with gr.Row():
@@ -85,6 +80,8 @@ with block:
     state = gr.State()
     agent_state = gr.State()
 
+    block.load(load_model, inputs=None, outputs=[agent_state]
+
     submit.click(chat, inputs=[message, state, agent_state], outputs=[chatbot, state])
     message.submit(chat, inputs=[message, state, agent_state], outputs=[chatbot, state])