Update app.py

app.py

@@ -1,19 +1,18 @@
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import AutoModelForSeq2SeqLM,AutoTokenizer
 import gradio as grad
 
-model_name = "Helsinki-NLP/opus-mt-ko-en"
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+mdl_name = "Helsinki-NLP/opus-mt-ko-en"
+mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
+my_tkn = AutoTokenizer.from_pretrained(mdl_name)
 
-#opus_translator = pipeline("translation", model=model_name)
+#opus_translator = pipeline("translation", model=mdl_name)
 
 def translate(text):
-  inputs = tokenizer(text, return_tensors="pt")
-  translation_output = model.generate(**input)
-  response = tokenizer.decode(translation_output[0], skip_special_tokens=True)
+    inputs = my_tkn(text, return_tensors="pt")
+    trans_output = mdl.generate(**inputs)
+    response = my_tkn.decode(trans_output[0], skip_special_tokens=True)
 
-  #response = opus_translator(text)
-  return response
+    #response = opus_translator(text)
+    return response
 
-# Web UI
-grad.Interface(translate, inputs=["text"], outputs=["text"]).launch()
+grad.Interface(translate, inputs=["text",], outputs="text").launch()