Munzali commited on
Commit
151cf50
·
verified ·
1 Parent(s): 1f00609

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -62,7 +62,8 @@ stored_imgs = load_stored_images()
62
  @spaces.GPU
63
  def check_membership(uploaded_image):
64
  try:
65
- image = cv2.resize(uploaded_image, (256, 256))
 
66
 
67
  uploaded_image = np.expand_dims(image, axis=0).astype("float32")
68
 
 
62
  @spaces.GPU
63
  def check_membership(uploaded_image):
64
  try:
65
+ cv_img = cv2.cvtColor(np.array(uploaded_image), cv2.COLOR_RGB2BGR)
66
+ image = cv2.resize(cv_img, (256, 256))
67
 
68
  uploaded_image = np.expand_dims(image, axis=0).astype("float32")
69