fix

app.py

@@ -99,7 +99,7 @@ with block:
         cv2.circle(prompt_image, (x, y), 5, color, -1)
         point_coords.append([x,y])
         point_labels.append(point_label_radio)
-        sam_masks = sam_cpu.cond_pred(pts=np.array(point_coords), lbls=np.array(point_labels), embedding=embedding, image_size=input_image.size)
+        sam_masks = sam_cpu.cond_pred(pts=np.array(point_coords), lbls=np.array(point_labels), embedding=embedding)
         return  [ prompt_image,
                   (input_image, sam_masks),
                   point_coords,

inference.py

@@ -232,7 +232,6 @@ class CustomSamPredictor(SamPredictor):
 
     def decode_and_predict(
         self,
-        image_size: Tuple[int, int],
         embedding: torch.Tensor,
         point_coords: Optional[np.ndarray] = None,
         point_labels: Optional[np.ndarray] = None,