ybelkada HF staff commited on
Commit
7298a6e
1 Parent(s): 1818f9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def inference(raw_image, question, decoding_strategy):
34
  inputs = [
35
  gr.inputs.Image(type='pil'),
36
  gr.inputs.Textbox(lines=2, label="Context (optional)"),
37
- gr.inputs.Radio(choices=['Beam search','Nucleus sampling'], type="value", default="Nucleus sampling", label="Caption Decoding Strategy")
38
  ]
39
  outputs = gr.outputs.Textbox(label="Output")
40
 
34
  inputs = [
35
  gr.inputs.Image(type='pil'),
36
  gr.inputs.Textbox(lines=2, label="Context (optional)"),
37
+ gr.inputs.Radio(choices=['Beam search','Nucleus sampling', 'Contrastive search'], type="value", default="Nucleus sampling", label="Caption Decoding Strategy")
38
  ]
39
  outputs = gr.outputs.Textbox(label="Output")
40