Update app.py

app.py

@@ -8,11 +8,16 @@ model_name = "Helsinki-NLP/opus-mt-en-es"
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 model = MarianMTModel.from_pretrained(model_name)
 
+def traducir_resumir(texto):
+    text_summarized = summarizer(texto, max_length=1000, min_length=30, do_sample=False)
+    texto_solo = ">>esp<< " + text_summarized[0]["summary_text"]
+    translated_resumido = model.generate(**tokenizer(texto_solo, return_tensors="pt", padding=True))
+    texto_resumido = [tokenizer.decode(t, skip_special_tokens=True) for t in translated_resumido]
+    return texto_resumido[0]
+
 def traducir(texto):
     texto_unido = ""
     texto_completo = ""
-    text_summarized = summarizer(texto, max_length=1000, min_length=30, do_sample=False)
-    texto_solo = ">>esp<< " + text_summarized[0]["summary_text"]
     texto_separado = texto.split(" ")
     for i in range(len(texto_separado)):
         texto_unido = texto_unido + "".join(texto_separado[i]) + " "
@@ -21,16 +26,16 @@ def traducir(texto):
             texto_unido = [tokenizer.decode(t, skip_special_tokens=True) for t in texto_unido_traducido]
             texto_completo = texto_completo + "".join(texto_unido[0]) + " "    
             texto_unido = ""
-    translated_resumido = model.generate(**tokenizer(texto_solo, return_tensors="pt", padding=True))
-    texto_resumido = [tokenizer.decode(t, skip_special_tokens=True) for t in translated_resumido]
-    return texto_resumido[0], texto_completo
+    return texto_completo
 
 with gr.Blocks() as demo:
     text = gr.Textbox(label="Introduzca el texto a resumir y traducir")
-    output1 = gr.Textbox(label="Texto resumido y traducido")
-    output2 = gr.Textbox(label="Texto completo y traducido")
+    output1 = gr.Textbox(label="Texto resumido traducido")
+    output2 = gr.Textbox(label="Texto completo traducido")
     sumbtn = gr.Button("Resumir y traducir")
-    sumbtn.click(fn=traducir, inputs = text, outputs = [output1,output2])
+    tradbtn = gr.Button("Traducir texto")
+    sumbtn.click(fn = traducir_resumir, inputs = text, outputs = output1)
+    tradbtn.click(fn = traducir, inputs = text, outputs = output2)
 # demo = gr.Interface(
 #     fn=traducir,
 #     inputs=[gr.Textbox(label="Texto")],