Update app.py

app.py

@@ -3,13 +3,31 @@ from transformers import pipeline
 
 get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
+def translate(input_text, source, target):
+    # source_readable = source
+    # if source == "Auto Detect" or source.startswith("Detected"):
+    #   source, _ = auto_detect_language_code(input_text)
+    # if source in source_lang_dict.keys():
+    #   source = source_lang_dict[source]
+    # target_lang_dict, _ = get_target_languages(source)
+    try:
+      # target = target_lang_dict[target]
+      model = f"Helsinki-NLP/opus-mt-{source}-{target}"
+      pipe = pipeline("translation", model=model)
+      translation = pipe(input_text)
+      return translation[0]['translation_text'], ""
+    except KeyError:
+      return "", f"Error: Translation direction {source_readable} to {target} is not supported by Helsinki Translation Models"
+
 def summarize(input):
     output = get_completion(input)
-    return output[0]['summary_text']
+    summary_origin = output[0]['summary_text']
+    summary_translated = translate(summary_origin,'en','zh')
+    return summary_origin, summary_translated
 
 demo = gr.Interface(fn=summarize, 
                     inputs=[gr.Textbox(label="Text to summarize", lines=6)],
-                    outputs=[gr.Textbox(label="Result", lines=3)],
+                    outputs=[gr.Textbox(label="Result", lines=3),gr.Textbox(label="Translate Result", lines=3)],
                     title="Text summarization with distilbart-cnn",
                     description="Summarize any text using the `sshleifer/distilbart-cnn-12-6` model under the hood!",
                     examples=[