Update app.py

app.py

@@ -22,12 +22,12 @@ def get_caption(img):
 
 # 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]
+# 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()
@@ -48,10 +48,10 @@ 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)
+    # 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()