diff --git a/caikit_huggingface_demo/client/text_generation.py b/caikit_huggingface_demo/client/text_generation.py index e2d2bd0..c6c6d9c 100644 --- a/caikit_huggingface_demo/client/text_generation.py +++ b/caikit_huggingface_demo/client/text_generation.py @@ -45,12 +45,17 @@ def optional_tab(cls, models, request, predict): label="Model ID", choices=models, value=models[0] ) inputs = gr.Textbox( - label="Input Text", placeholder=f"Enter input text for {tab}" + label='Input Text', placeholder=f'Prompt Text (hit enter to send)' ) outputs = gr.Textbox( label="Output Text", placeholder=f"Output text for {tab}" ) - inputs.change(this.fn, [model_choice, inputs], outputs, api_name=tab) + inputs.submit(this.fn, [model_choice, inputs], outputs, api_name=tab) + clear = gr.Button("Clear") + + def nones(*args): + return [None for x in args] + clear.click(nones, [inputs, outputs], [inputs, outputs], queue=False) model_choice.change( this.fn, [model_choice, inputs], outputs, api_name=tab )