Update app.py

app.py

@@ -3,11 +3,28 @@ from transformers import pipeline
 
 playground = gr.Blocks()
 image_pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
+
 
 def launch_image_pipe(input):
     out = image_pipe(input)
     return out[0]['generated_text']
 
+def translate(input_text, source, target):
+    try:
+      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)
+    summary_origin = output[0]['summary_text']
+    summary_translated = translate(summary_origin,'en','fr')
+    return summary_origin, summary_translated[0]
+
 def create_playground_header():
     gr.Markdown("""
                 # 🤗 Hugging Face Labs
@@ -50,7 +67,7 @@ with playground:
                             "model: [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base)"]
             image_pipeline_button = create_tabs_header(topic, description, references)
             
-            with gr.Row(visible=True) as use_pipeline:
+            with gr.Row():
                 with gr.Column():
                     img = gr.Image(type='pil')
                 with gr.Column():
@@ -64,7 +81,30 @@ with playground:
             gr.Markdown("""
                         > Text Summarization and Translation
                         """)
-        
+            with gr.Row():
+                with gr.Column(scale=4):
+                    gr.Markdown("""
+                                ## Text Summarization and Translation
+                                ### Summarize the paragraph and translate it into other language.
+                                > pipeline: summarization, model: [sshleifer/distilbart-cnn-12-6](https://huggingface.co/sshleifer/distilbart-cnn-12-6)
+                                > pipeline: translation, model: [Helsinki-NLP/opus-mt-en-fr](https://huggingface.co/Helsinki-NLP/opus-mt-en-fr)
+                                """)
+                    
+                with gr.Column(scale=1):
+                    text_pipeline_button = gr.Button(value="Process")
+
+            with gr.Row():
+                with gr.Column():
+                    source_text = gr.Textbox(label="Text to summarize", lines=6)
+                with gr.Column():
+                    summary_textbox = gr.Textbox(lines=3, placeholder="", label="Summarization")
+                    translated_textbox = gr.Textbox(lines=3, placeholder="", label="Translate Result")
+
+                    
+            text_pipeline_button.click(summarize,
+                                        inputs=[source_text],
+                                        outputs=[summary_textbox, translated_textbox])
+            
         with gr.TabItem("Name Entity"):
             gr.Markdown("""
                         > Name Entity Recognition