Final updates

app.py

@@ -20,16 +20,6 @@ def get_caption(img):
     output = caption_model(img)
     return output[0]["generated_text"]
 
-# 4. Image Generation
-# img_model = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
-# img_model = DiffusionPipeline.from_pretrained("Linaqruf/anything-v3.0")
-# def get_img(prompt):
-#     print("Here inside")
-#     img = img_model(prompt)
-#     print("Generated")
-#     return img.images[0]
-    
-
 demo = gr.Blocks()
 with demo:
     gr.Markdown("# Try out some cool tasks!")
@@ -48,10 +38,5 @@ with demo:
         cap_btn = gr.Button("Generate Caption")
         cap_output = [gr.Textbox(label="Caption")]
         cap_btn.click(get_caption, cap_input, cap_output)
-    # with gr.Tab("Image Generation"):
-    #     img_input = [gr.Textbox(label="Your Text")]
-    #     img_btn = gr.Button("Generate Image")
-    #     img_output = [gr.Image(label="Generated Image")]
-    #     img_btn.click(get_img, img_input, img_output)
 
 demo.launch()