Update app.py

app.py

@@ -30,8 +30,14 @@ model, processor = get_vision_model()
 # Function to get image caption via Kosmos2.
 @st.cache_data
 def get_image_caption(image_data):
-    # Convert BytesIO to PIL Image
-    image = Image.open(io.BytesIO(image_data))
+    # Ensure image_data is a bytes stream ready to be read by Image.open
+    if isinstance(image_data, io.BytesIO):
+        # If it's already a BytesIO, we need to seek to the beginning of the file
+        image_data.seek(0)
+        image = Image.open(image_data)
+    else:
+        # If image_data is not a BytesIO object, create one
+        image = Image.open(io.BytesIO(image_data.read()))
 
     model, processor = get_vision_model()