Application file

app.py

@@ -82,7 +82,7 @@ def generate_image_with_sam(img, boxes):
     
     image_embeddings = model_sam.get_image_embeddings(inputs["pixel_values"])
 
-    inputs = processor_sam(img_pil, input_boxes=boxes, return_tensors="pt").to(device)
+    inputs = processor_sam(img_pil, input_boxes=[boxes], return_tensors="pt").to(device)
     inputs["input_boxes"].shape
     inputs.pop("pixel_values", None)
     inputs.update({"image_embeddings": image_embeddings})
@@ -91,10 +91,7 @@ def generate_image_with_sam(img, boxes):
         outputs = model_sam(**inputs, multimask_output=False)
     
     masks = processor_sam.image_processor.post_process_masks(outputs.pred_masks.cpu(), inputs["original_sizes"].cpu(), inputs["reshaped_input_sizes"].cpu())
-    scores = outputs.iou_scores
-    print(type(scores))
-    print(scores.shape if hasattr(scores, 'shape') else scores)
-
+    # scores = outputs.iou_scores
 
     SAM_image = apply_colored_masks_on_image(img_pil, masks[0])
     return SAM_image