Spaces:

not-lain
/

gpu-utils

Paused

not-lain commited on Jan 17, 2025

Commit

430a7a9

1 Parent(s): 9dbd7e8

switch to sam2 large

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,7 +125,7 @@ def rmbg(image=None, url=None):
 def mask_generation(image=None, d=None):
     d = eval(d)  # convert this to dictionary
-    predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-tiny")
     predictor.set_image(image)
     input_point = np.array(d["input_points"])
     input_label = np.array(d["input_labels"])

 def mask_generation(image=None, d=None):
     d = eval(d)  # convert this to dictionary
+    predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2.1-hiera-large")
     predictor.set_image(image)
     input_point = np.array(d["input_points"])
     input_label = np.array(d["input_labels"])