Update app.py

app.py

@@ -1,31 +1,49 @@
 import gradio as gr
+from transformers import pipeline
 
 playground = gr.Blocks()
+image_pipe = pipeline("image-to-text",
+                model="Salesforce/blip-image-captioning-base")
 
 def create_playground_header():
     gr.Markdown("""
-                🤗 Hugging Face Labs
-                ** Explore different LLM on Hugging Face platform. Just play and enjoy **
+                #🤗 Hugging Face Labs
+                **Explore different LLM on Hugging Face platform. Just play and enjoy**
                 """)
 
 def create_playground_footer():
     gr.Markdown("""
-                ### To Learn More about 🤗 Hugging Face,[Click Here](https://huggingface.co/docs)
+                **To Learn More about 🤗 Hugging Face, [Click Here](https://huggingface.co/docs)**
                 """)
 
+def create_tabs_header(description):
+    with gr.Row():
+        with gr.Column(scale=4):
+            gr.Markdown(f"**{description}**")
+        with gr.Column(scale=1):
+            test_pipeline_button = gr.Button(value="Process")
+        return test_pipeline_button
+
 with playground:
     create_playground_header()
     with gr.Tabs():
+        with gr.TabItem("Image"):
+            image_pipeline_button = create_tabs_header("Image Captioning")
+            with gr.Row(visible=True) as use_pipeline:
+                with gr.Column():
+                    img = gr.Image(type='pil')
+                with gr.Column():
+                    generated_textbox = gr.Textbox(lines=2, placeholder="", label="Generated Text")
+                    
+            image_pipeline_button.click(image_pipe,
+                                        inputs=[img],
+                                        outputs=[generated_textbox]])
+            
         with gr.TabItem("Text"):
             gr.Markdown("""
                         > Text Summarization and Translation
                         """)
         
-        with gr.TabItem("Image"):
-            gr.Markdown("""
-                        > Image Captioning
-                        """)
-            
         with gr.TabItem("Name Entity"):
             gr.Markdown("""
                         > Name Entity Recognition