Update app.py

app.py

@@ -1,31 +1,17 @@
 import gradio as gr
 from transformers import pipeline
-from diffusers import StableDiffusionPipeline
 
 playground = gr.Blocks()
 
 image_pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
 summary_pipe = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 ner_pipe = pipeline("ner", model="dslim/bert-base-NER")
-pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+
 
 def launch_image_pipe(input):
     out = image_pipe(input)
     return out[0]['generated_text']
-
-def base64_to_pil(img_base64):
-    base64_decoded = base64.b64decode(img_base64)
-    byte_stream = io.BytesIO(base64_decoded)
-    pil_image = Image.open(byte_stream)
-    return pil_image
-
-def image_generate(prompt):
-    # output = get_completion(prompt)
-    image = pipe(prompt).images[0]
-    result_image = base64_to_pil(image)
-    return result_image
-    
+   
 def translate(input_text, source, target):
     try:
       model = f"Helsinki-NLP/opus-mt-{source}-{target}"
@@ -81,35 +67,53 @@ def create_playground_footer():
                 **To Learn More about 🤗 Hugging Face, [Click Here](https://huggingface.co/docs)**
                 """)
 
-def create_tabs_header(topic, description, references):
-    with gr.Row():
-        with gr.Column(scale=4):
-            # reference_list = "> " + "\n> ".join(references)
-            # content  = f"## {topic}\n"
-            # content += f"### {description}\n"
-            # for ref in references:
-            #     content += f"> {ref}\n"
-            # gr.Markdown(content)
-            gr.Markdown("""
-                        ## Image Captioning
-                        ### Upload a image, check what AI understand and have vision on it.
-                        > category: Image-to-Text, model: [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base)
-                        """)
+# def create_tabs_header(topic, description, references):
+#     with gr.Row():
+#         with gr.Column(scale=4):
+#             # reference_list = "> " + "\n> ".join(references)
+#             # content  = f"## {topic}\n"
+#             # content += f"### {description}\n"
+#             # for ref in references:
+#             #     content += f"> {ref}\n"
+#             # gr.Markdown(content)
+#             gr.Markdown("""
+#                         ## Image Captioning
+#                         ### Upload a image, check what AI understand and have vision on it.
+#                         > category: Image-to-Text, model: [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base)
+#                         """)
             
-        with gr.Column(scale=1):
-            test_pipeline_button = gr.Button(value="Start Process", variant="primary")
-        return test_pipeline_button
+#         with gr.Column(scale=1):
+#             test_pipeline_button = gr.Button(value="Start Process", variant="primary")
+#         return test_pipeline_button
 
 with playground:
     create_playground_header()
     with gr.Tabs():
         with gr.TabItem("Image"):
             
-            topic = "Image Captioning"
-            description = "Upload a image, check what AI understand and have vision on it."
-            references = ["category: Image-to-Text",
-                            "model: [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base)"]
-            image_pipeline_button = create_tabs_header(topic, description, references)
+            # topic = "Image Captioning"
+            # description = "Upload a image, check what AI understand and have vision on it."
+            # references = ["category: Image-to-Text",
+            #                 "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():
+                with gr.Column(scale=4):
+                    # reference_list = "> " + "\n> ".join(references)
+                    # content  = f"## {topic}\n"
+                    # content += f"### {description}\n"
+                    # for ref in references:
+                    #     content += f"> {ref}\n"
+                    # gr.Markdown(content)
+                    gr.Markdown("""
+                                ## Image Captioning
+                                ### Upload a image, check what AI understand and have vision on it.
+                                > category: Image-to-Text, model: [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base)
+                                """)
+            
+                with gr.Column(scale=1):
+                    ITT_button = gr.Button(value="Start Process", variant="primary")
+                    ITT_Clear_button = gr.ClearButton(components=[img, generated_textbox], value="Clear")
             
             with gr.Row():
                 with gr.Column():
@@ -123,8 +127,8 @@ with playground:
                     with gr.Row():
                         generated_image = gr.Image(label="Generated Image")
                         
-            image_pipeline_button.click(launch_image_pipe, inputs=[img], outputs=[generated_textbox])
-            image_generation_button.click(image_generate, inputs=[generated_textbox], outputs=[generated_image])
+            ITT_button.click(launch_image_pipe, inputs=[img], outputs=[generated_textbox])
+            # image_generation_button.click(image_generate, inputs=[generated_textbox], outputs=[generated_image])
         
         with gr.TabItem("Text"):
             with gr.Row():