subhc commited on
Commit
0ade859
1 Parent(s): fa11065

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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,