Update app.py

app.py

@@ -2,23 +2,28 @@
 # Autor..: joinfv
 # Data...: 20/11/2024
 import gradio as gr
-
+import json
 from transformers import pipeline
 
+# title = "Tradutor inglês para brasileiro"
+
 # Load model
 pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-en-pt")
 
-def predict(text):
-    return pipe(text)[0]["translation_text"]
+def translate(text):
+    result = pipe(text)[0]["translation_text"]
+    return {"translation": result}
+#    return pipe(text)[0]["translation_text"]
+
 
-title = "Tradutor inglês para brasileiro"
 
-iface = gr.Interface(
-    fn=predict,  # Passa a referência à função predict
-    inputs=[gr.Textbox(label="text", lines=3)],
-    outputs='text',
-    title=title,
-    api_name="predict"
+demo = gr.Interface(
+    fn=translate,
+    inputs="text",
+    outputs="text",
+    title="Tradutor Inglês-Português",
+    description="Traduza textos do inglês para o português"
 )
 
-iface.launch()
+if __name__ == "__main__":
+    demo.launch()