Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,7 +62,8 @@ stored_imgs = load_stored_images()
|
|
| 62 |
@spaces.GPU
|
| 63 |
def check_membership(uploaded_image):
|
| 64 |
try:
|
| 65 |
-
|
|
|
|
| 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 |
|