Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -100,13 +100,9 @@ def cropfromsizedivisible(img, size_divisibility, orig_size):
|
|
100 |
def evaluate_image(image_path):
|
101 |
binary_threshold = 0.5
|
102 |
metadata = MetadataCatalog.get("__unused")
|
103 |
-
|
104 |
-
model, cfg = load_model_cfg("DAVIS")
|
105 |
|
106 |
image_pil = Image.open(image_path).convert('RGB')
|
107 |
-
|
108 |
image_pil.thumbnail((384, 384))
|
109 |
-
|
110 |
osize = image_pil.size
|
111 |
if cfg.MODEL.MASK_FORMER.SIZE_DIVISIBILITY > 0:
|
112 |
image_pil = expand2sizedivisible(image_pil, 0, cfg.MODEL.MASK_FORMER.SIZE_DIVISIBILITY)
|
@@ -144,6 +140,8 @@ def evaluate_image(image_path):
|
|
144 |
|
145 |
paths = sorted(pathlib.Path('samples').glob('*.jpg'))
|
146 |
css = ".component-1 {height: 256px !important;}"
|
|
|
|
|
147 |
merger = MaskMerger(cfg, model, merger_model="dino_vitb8")
|
148 |
demo = gr.Interface(
|
149 |
fn=evaluate_image,
|
|
|
100 |
def evaluate_image(image_path):
|
101 |
binary_threshold = 0.5
|
102 |
metadata = MetadataCatalog.get("__unused")
|
|
|
|
|
103 |
|
104 |
image_pil = Image.open(image_path).convert('RGB')
|
|
|
105 |
image_pil.thumbnail((384, 384))
|
|
|
106 |
osize = image_pil.size
|
107 |
if cfg.MODEL.MASK_FORMER.SIZE_DIVISIBILITY > 0:
|
108 |
image_pil = expand2sizedivisible(image_pil, 0, cfg.MODEL.MASK_FORMER.SIZE_DIVISIBILITY)
|
|
|
140 |
|
141 |
paths = sorted(pathlib.Path('samples').glob('*.jpg'))
|
142 |
css = ".component-1 {height: 256px !important;}"
|
143 |
+
|
144 |
+
model, cfg = load_model_cfg("DAVIS")
|
145 |
merger = MaskMerger(cfg, model, merger_model="dino_vitb8")
|
146 |
demo = gr.Interface(
|
147 |
fn=evaluate_image,
|