Update app.py

app.py

@@ -71,7 +71,7 @@ def inference_mask1(
         "query_img": query_img,
     }
     nshot = len(support_masks)
-    pred_mask, simi, simi_map = model.module.predict_mask_nshot(batch, nshot=nshot)
+    pred_mask, simi, simi_map = model.predict_mask_nshot(batch, nshot=nshot)
     pred_mask = F.interpolate(pred_mask, shape)
     pred_mask = pred.detach().cpu().numpy()
     output_img = query_img_np * 0.5 + 0.5 * np.array([1, 0, 0]) * np.expand_dims(pred_mask, axis=0)