Upload app.py

app.py

@@ -48,17 +48,17 @@ def infer(message: str, sysprompt: str, tokens: int=30):
     return output_md, info_md
 
 with gr.Blocks() as demo:
+    model_name = gr.Dropdown(label="Model", choices=MODELS, value=MODELS[0], allow_custom_value=True)
     with gr.Row():
         message = gr.Textbox(label="Message", value="", lines=1)
         sysprompt = gr.Textbox(label="System prompt", value="You are Qwen, created by Alibaba Cloud. You are a helpful assistant.", lines=4)
         tokens = gr.Slider(label="Max tokens", value=30, minimum=1, maximum=2048, step=1)
-        model_name = gr.Dropdown(label="Model", choices=MODELS, value=MODELS[0], allow_custom_value=True)
-        #image_url = gr.Textbox(label="Image URL", value=url, lines=1)
     run_button = gr.Button("Run", variant="primary")
     output_md = gr.Markdown("<br><br>")
     info_md = gr.Markdown("<br><br>")
 
-    run_button.click(infer, [message, sysprompt, tokens], [output_md, info_md])
+    gr.on(triggers=[run_button.click, message.submit], fn=infer, inputs=[message, sysprompt, tokens], outputs=[output_md, info_md])
+    #run_button.click(infer, [message, sysprompt, tokens], [output_md, info_md])
     model_name.change(load_model, [model_name], [model_name])
 
 demo.launch()