Update app.py

app.py

@@ -1,8 +1,18 @@
 import gradio as gr
+from transformers import AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer
 
-prefix = "translate Spanish to Nahuatl: "
-gr.Interface.load(
-  "huggingface/hackathon-pln-es/t5-small-spanish-nahuatl",
+model = AutoModelForSeq2SeqLM.from_pretrained('hackathon-pln-es/t5-small-spanish-nahuatl')
+tokenizer = AutoTokenizer.from_pretrained('hackathon-pln-es/t5-small-spanish-nahuatl')
+
+def predict(input):
+  input_ids = tokenizer('translate Spanish to Nahuatl: ' + sentence, return_tensors='pt').input_ids
+  outputs = model.generate(input_ids)
+  outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
+  return outputs
+
+gr.Interface(
+   fn=predict,
    inputs=gr.inputs.Textbox(lines=1, label="Input Text"),
    theme="peach",
    title='🌽 Spanish to Nahuatl Automatic Translation',