Initial Commit

app.py

@@ -26,7 +26,9 @@ def inference(image, input_label, model_choice):
     sam.to(device=device)
     predictor = SamPredictor(sam)
     predictor.set_image(image)
-    input_point = np.array([[int(input_label['label'].split(',')[0]), int(input_label['label'].split(',')[1])]])
+    x = int(input_label['label'].split(',')[0])
+    y = int(input_label['label'].split(',')[1])
+    input_point = np.array([[x, y]])
     input_label = np.array([1])
     masks, scores, logits = predictor.predict(
                                                 point_coords=input_point,
@@ -36,9 +38,7 @@ def inference(image, input_label, model_choice):
     mask = masks[0]
     image2 = image.copy()
     image2[mask, 0] = 255
-    image2[int(input_label['label'].split(',')[0])-10:int(input_label['label'].split(',')[0])+10,
-           int(input_label['label'].split(',')[1])-10:int(input_label['label'].split(',')[1])+10,
-           2] = 255
+    image2[x-10:x+10, y-10:y+10, 2] = 255
     return image2