Sifal commited on
Commit
55fb831
1 Parent(s): 23cebfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -258,8 +258,8 @@ iface = gr.Interface(
258
  inputs=[
259
  gr.Textbox(label="Input sentence"),
260
  gr.Radio(['greedy', 'beam search'], label="Decoding Strategy"),
261
- gr.Number(value=5,label="Length Extend (for greedy)",precision=0),
262
- gr.Number(value=1,label="Beam Size (for beam search)",precision=0),
263
  gr.Number(value=0.6,label="Length Penalty (for beam search)")
264
  ],
265
  outputs=gr.Textbox(label="Translation"),
 
258
  inputs=[
259
  gr.Textbox(label="Input sentence"),
260
  gr.Radio(['greedy', 'beam search'], label="Decoding Strategy"),
261
+ gr.Number(value=5,label="Max tokens",precision=0),
262
+ gr.Number(value=5,label="Beam Size (for beam search)",precision=0),
263
  gr.Number(value=0.6,label="Length Penalty (for beam search)")
264
  ],
265
  outputs=gr.Textbox(label="Translation"),