Update app.py

app.py

@@ -79,5 +79,15 @@ neutralizer = pipeline('text2text-generation', model='islasher/mbart-spanishToQu
 #     trad = tokenizer.decode(outputs[0], skip_special_tokens=True)
 #     return trad
 
+
+
+def predict(frase):
+#     inputs = tokenizer(frase, return_tensors="pt")
+#     outputs = model(**inputs)
+#     trad = tokenizer.decode(outputs[0], skip_special_tokens=True)
+     return neutralizer['generated_text']
+
+# Creamos la interfaz y la lanzamos. 
+gr.Interface(fn=neutralizer, inputs="text", outputs="text").launch(share=False)
 # Creamos la interfaz y la lanzamos. 
 gr.Interface(fn=neutralizer, inputs="text", outputs="text").launch(share=False)