jfeketet72 commited on
Commit
32f5401
1 Parent(s): 9045e57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -1,5 +1,4 @@
1
- from transformers import AutoModel,AutoTokenizer,AutoModelFor
2
- Seq2SeqLM
3
  import gradio as grad
4
  mdl_name = "Helsinki-NLP/opus-mt-en-hu"
5
  mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
@@ -11,6 +10,4 @@ def translate(text):
11
  response = my_tkn.decode(trans_output[0], skip_special_tokens=True)
12
  #response = opus_translator(text)
13
  return response
14
- txt=grad.Textbox(lines=1, label="English", placeholder="English Text here")
15
- out=grad.Textbox(lines=1, label="Hungarian")
16
- grad.Interface(translate, inputs=txt, outputs=out).launch()
 
1
+ from transformers import AutoModelForSeq2SeqLM,AutoTokenizer
 
2
  import gradio as grad
3
  mdl_name = "Helsinki-NLP/opus-mt-en-hu"
4
  mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
 
10
  response = my_tkn.decode(trans_output[0], skip_special_tokens=True)
11
  #response = opus_translator(text)
12
  return response
13
+ grad.Interface(translate, inputs=["text",], outputs="text").launch()