update window
Browse files
app.py
CHANGED
|
@@ -15,7 +15,6 @@ birefnet = AutoModelForImageSegmentation.from_pretrained(
|
|
| 15 |
)
|
| 16 |
birefnet.to("cuda")
|
| 17 |
|
| 18 |
-
predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-tiny", device="cpu")
|
| 19 |
|
| 20 |
transform_image = transforms.Compose(
|
| 21 |
[
|
|
@@ -124,6 +123,7 @@ def rmbg(image=None, url=None):
|
|
| 124 |
|
| 125 |
|
| 126 |
def mask_generation(image=None, json=None):
|
|
|
|
| 127 |
return None
|
| 128 |
|
| 129 |
|
|
|
|
| 15 |
)
|
| 16 |
birefnet.to("cuda")
|
| 17 |
|
|
|
|
| 18 |
|
| 19 |
transform_image = transforms.Compose(
|
| 20 |
[
|
|
|
|
| 123 |
|
| 124 |
|
| 125 |
def mask_generation(image=None, json=None):
|
| 126 |
+
predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-tiny")
|
| 127 |
return None
|
| 128 |
|
| 129 |
|