added fix with png images

app.py

@@ -1,4 +1,5 @@
 import os
+import numpy as np
 import boto3
 import streamlit as st
 import faiss
@@ -6,6 +7,7 @@ import pandas as pd
 from PIL import Image
 from model_prediction import Ranker
 from io import BytesIO
+import cv2
 
 
 @st.cache
@@ -63,6 +65,13 @@ bucket_name = "product-retrieval"
 
 if uploaded_file is not None:
     image = Image.open(uploaded_file)
+    image = np.asarray(image)
+
+    if len(image.shape) > 2 and image.shape[2] == 4:
+        image = cv2.cvtColor(image, cv2.COLOR_BGRA2BGR)
+
+    image = Image.fromarray(image)
+
     st.image(image, caption="Uploaded image", use_column_width=True)
 
     loading_text.text("Loading predictions...")

requirements.txt

@@ -5,4 +5,5 @@ Pillow
 scikit-learn
 torchvision
 torch
-open_clip_torch
+open_clip_torch
+opencv-python