Update app.py

app.py

@@ -32,9 +32,9 @@ def add_text(history, text):
     history = history + [(text, None)]
     return history, gr.Textbox(value="", interactive=False)
 
-def bot(history):
+def bot(history, max_length=100, min_length=20, temperature=1.0):
     user_input = history[-1][0]
-    generated_response = generate_text(user_input)
+    generated_response = generate_text(user_input, max_length, min_length, temperature)
     history[-1][1] = generated_response
     time.sleep(0.5)
     yield history
@@ -45,6 +45,7 @@ with gr.Blocks() as demo:
         [],
         elem_id="chatbot",
         bubble_full_width=False,
+        avatar_images=(None, (os.path.join(os.path.dirname(__file__), "avatar.png"))),
     )
 
     with gr.Row():
@@ -55,8 +56,12 @@ with gr.Blocks() as demo:
             container=False,
         )
 
+        max_len_slider = gr.Slider(0, 2048, 100, label="Max Length")
+        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, chatbot, api_name="bot_response"
+        bot, chatbot, chatbot, max_len_slider, min_len_slider, temp_slider, api_name="bot_response"
     )
     txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)