Update app.py

app.py

@@ -1,12 +1,16 @@
-import gradio as gr
+from transformers import AutoModelForSeq2SeqLM,AutoTokenizer
+import gradio as grad
+mdl_name = "Helsinki-NLP/opus-mt-es-en"
+mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
+my_tkn = AutoTokenizer.from_pretrained(mdl_name)
 
-description = "Story generation with GPT-2"
-title = "Generate your own story"
-examples = [["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
+#opus_translator = pipeline("translation", model=mdl_name)
 
-interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
-            description=description,
-            examples=examples
-)
+def translate(text):
+    inputs = my_tkn(text, return_tensors="pt")
+    trans_output = mdl.generate(**inputs)
+    response = my_tkn.decode(trans_output[0], skip_special_tokens=True)
 
-interface.launch()
+    #response = opus_translator(text)
+    return response
+grad.Interface(translate, inputs=["text",], outputs="text").launch()