Update app.py

app.py

@@ -37,14 +37,14 @@ def bot(history, max_len, min_len, temp):
     generated_response = generate_text(user_input, max_length=max_len, min_length=min_len, temperature=temp)
     history[-1][1] = generated_response
     time.sleep(0.5)
-    yield history
-
+    return history
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot(
         [],
         elem_id="chatbot",
         bubble_full_width=False,
+        avatar_images=(None, (os.path.join(os.path.dirname(__file__), "avatar.png"))),
     )
 
     with gr.Row():
@@ -59,9 +59,9 @@ with gr.Blocks() as demo:
         min_len_slider = gr.Slider(0, 2048, 20, label="Min Length")
         temp_slider = gr.Slider(0.1, 2.0, 1.0, label="Temperature")
 
-    txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
-        bot, chatbot, max_len_slider, min_len_slider, temp_slider, api_name="bot_response"
-    )
+    txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False)
+    txt_msg.then(bot, chatbot, max_len_slider, min_len_slider, temp_slider)
+
     txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
 
     chatbot.like(print_like_dislike, None, None)
@@ -69,4 +69,3 @@ with gr.Blocks() as demo:
 demo.queue()
 if __name__ == "__main__":
     demo.launch()
-