Update app.py

app.py

@@ -24,7 +24,11 @@ def get_caption(img):
 # img_model = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
 img_model = DiffusionPipeline.from_pretrained("Linaqruf/anything-v3.0")
 def get_img(prompt):
-    return img_model(prompt).images[0]
+    print("Here inside")
+    img = img_model(prompt)
+    print("Generated")
+    return img.images[0]
+    
 
 demo = gr.Blocks()
 with demo:
@@ -36,8 +40,8 @@ with demo:
         sum_btn.click(get_summary, sum_input, sum_output)
     with gr.Tab("Named Entity Recognition"):
         ner_input = [gr.Textbox(label="Text to find Entities", placeholder = "Enter text...", lines = 4)]
-        ner_output = [gr.HighlightedText(label="Text with entities")]
         ner_btn = gr.Button("Generate entities")
+        ner_output = [gr.HighlightedText(label="Text with entities")]
         ner_btn.click(get_ner, ner_input, ner_output)
     with gr.Tab("Image Captioning"):
         cap_input = [gr.Image(label="Upload Image", type="pil")]