Update app.py

app.py

@@ -17,6 +17,15 @@ def read_image(uploaded_file):
     os.remove(uploaded_file.name)  # Remove the temporary file
     return img
 
+def image_open(image_file):
+    img = cv2.imread(
+        image_file, cv2.IMREAD_COLOR | cv2.IMREAD_IGNORE_ORIENTATION
+    )
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    if img is None:
+        raise ValueError('Failed to read {}'.format(image_file))
+    return img
+
     
 # Function to display images from their paths
 def display_images(image_ids):
@@ -25,7 +34,7 @@ def display_images(image_ids):
     for img_id in image_ids:
         image_path = df.loc[df['seller_img_id'] == img_id, 'img_path'].values[0]
         print(image_path)
-        image = read_image(image_path)
+        image = image_open(image_path)
         images.append(image)
         captions.append(img_id)
     return images, captions