CUHKWilliam commited on
Commit
753655d
·
verified ·
1 Parent(s): 8fb9092

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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