Spaces:

mestvnvo
/

Kosmos2-API

Running

mestvnvo commited on Jul 18

Commit

7ab2903

1 Parent(s): 2ef3d5b

empty image catch

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,8 +9,8 @@ pipe = pipeline("image-text-to-text",
     )
 def get_image_caption(image):
-    if image is None:
-        return "No image provided."
     image = image.convert("RGB")

     )
 def get_image_caption(image):
+    if not image:
+        raise gr.Error("No image provided.")
     image = image.convert("RGB")