Iosif24 commited on
Commit
f0124b9
·
1 Parent(s): cdaa05d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as grad
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
 
4
- model_name = "Helsini-NLP/opus-mt-en-de"
5
 
6
  model = AutoModelForSeq2SeqLM(model_name)
7
  model_tkn = AutoTokenizer(model_name)
@@ -14,4 +14,6 @@ def translate(text):
14
  response = model_tkn.decode(trans_output[0], skip_special_tokens=True)
15
  #response = opus_translator(text)
16
  return response
17
- grad.Interface(translate, inputs="text", output="text").launch()
 
 
 
1
  import gradio as grad
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
 
4
+ model_name = "Helsinki-NLP/opus-mt-en-de"
5
 
6
  model = AutoModelForSeq2SeqLM(model_name)
7
  model_tkn = AutoTokenizer(model_name)
 
14
  response = model_tkn.decode(trans_output[0], skip_special_tokens=True)
15
  #response = opus_translator(text)
16
  return response
17
+ txt = grad.Textbox(line=1, label="English", placeholder = "English text here")
18
+ txt_out = grad.Textbox(line=1, label= "German", placeholder = "German translation")
19
+ grad.Interface(translate, inputs=txt, output=txt_out, theme="dark").launch()