Update app.py

app.py

@@ -32,13 +32,14 @@ def create_captions_rich(images):
     """
     Generates captions for input images.
     Args:
-        images (list): List of images to generate captions for.
+        images (list): List of image file paths to generate captions for.
     Returns:
         list: List of captions, one for each input image.
     """
     captions = []
-    for image in images:
+    for image_path in images:
         try:
+            image = Image.open(image_path).convert("RGB")
             prompt = "caption en"
             model_inputs = processor(text=prompt, images=image, return_tensors="pt").to("cpu")
             input_len = model_inputs["input_ids"].shape[-1]