Spaces:
Runtime error
Runtime error
bthndmn12
commited on
Commit
•
756b28b
1
Parent(s):
f000144
fixed some bugs
Browse files
app.py
CHANGED
|
@@ -35,8 +35,8 @@ def get_bbox(gt_map):
|
|
| 35 |
|
| 36 |
|
| 37 |
def greet(image):
|
| 38 |
-
|
| 39 |
-
image = image.resize((256, 256))
|
| 40 |
|
| 41 |
gt_mask = np.array(image)
|
| 42 |
prompt = get_bbox(gt_mask)
|
|
|
|
| 35 |
|
| 36 |
|
| 37 |
def greet(image):
|
| 38 |
+
image = Image.fromarray(image)
|
| 39 |
+
image = image.resize((256, 256))
|
| 40 |
|
| 41 |
gt_mask = np.array(image)
|
| 42 |
prompt = get_bbox(gt_mask)
|