Update app.py

app.py

@@ -50,25 +50,12 @@ def caption_random_image():
 demo = gr.Interface(
     fn=caption_random_image,
     inputs=None,
-    outputs=["image", "text"],
-    title="Image Captioning",
-    description="Pulls a random image from merve/coco and captions it using BLIP.",
+    outputs=[gr.Image(type="pil"), gr.Textbox(label="Caption")],
+    title="Random COCO Image Captioning",
+    description="Selects a random COCO image from 20 samples and generates a BLIP caption."
 )
 
 
-# conditional image captioning
-text = "a photography of"
-inputs = processor(image, text, return_tensors="pt")
-
-out = model.generate(**inputs)
-print(processor.decode(out[0], skip_special_tokens=True))
-
-# unconditional image captioning
-inputs = processor(image, return_tensors="pt")
-
-out = model.generate(**inputs)
-print(processor.decode(out[0], skip_special_tokens=True))
-
 
 demo.launch()