jin-nin commited on
Commit
6bae264
Β·
1 Parent(s): 3e2a2f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -18,12 +18,13 @@ def translate(lang, text):
18
  generated_tokens = model.generate(**encoded_text)
19
  return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
20
 
21
- Output = gr.outputs.Textbox()
 
 
 
22
  gr.Interface(
23
  translate,
24
- inputs=[
25
- gr.inputs.Dropdown( lang_list, label="To Language" ),
26
- 'text'
27
- ],
28
- outputs=[ Output ],
29
  ).launch()
 
18
  generated_tokens = model.generate(**encoded_text)
19
  return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
20
 
21
+ Language = gr.Dropdown( lang_list, label="Language" )
22
+ Source = gr.Textbox( lang_list, label="Source" )
23
+ Result = gr.Textbox( label="Result" )
24
+
25
  gr.Interface(
26
  translate,
27
+ inputs=[ Source, Language ],
28
+ outputs=[ Result ],
29
+ api_name="translate",
 
 
30
  ).launch()