Hugging Face's logo

Spaces:
CUHKWilliam
/
Multiple-shots
Sleeping

App Files Community
CUHKWilliam commited on
Commit
79d8b10
·
verified ·
1 Parent(s): f79acf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ def inference_mask1(
74
  "support_imgs": support_img,
75
  "query_img": query_img,
76
  }
77
- nshot = len(support_masks)
78
  pred_mask, simi, simi_map = model.predict_mask_nshot(batch, nshot=nshot)
79
  pred_mask = F.interpolate(pred_mask, shape)
80
  pred_mask = pred.detach().cpu().numpy()
 
74
  "support_imgs": support_img,
75
  "query_img": query_img,
76
  }
77
+ nshot = support_masks.size(1)
78
  pred_mask, simi, simi_map = model.predict_mask_nshot(batch, nshot=nshot)
79
  pred_mask = F.interpolate(pred_mask, shape)
80
  pred_mask = pred.detach().cpu().numpy()