Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ def inference_mask1(
|
|
| 24 |
query_img,
|
| 25 |
*prompt,
|
| 26 |
):
|
| 27 |
-
query_img = Image(query_img)
|
| 28 |
query_img_np = np.asarray(query_img)
|
| 29 |
query_img = transformation(query_img)
|
| 30 |
shape = query_img_np.shape
|
|
|
|
| 24 |
query_img,
|
| 25 |
*prompt,
|
| 26 |
):
|
| 27 |
+
query_img = Image.fromarray(query_img)
|
| 28 |
query_img_np = np.asarray(query_img)
|
| 29 |
query_img = transformation(query_img)
|
| 30 |
shape = query_img_np.shape
|