poptree commited on
Commit
6da3f2e
·
verified ·
1 Parent(s): cd05786

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -88,8 +88,8 @@ def run_model(target_dir, model, anchor_size=100) -> dict:
88
  # relocalization on all images
89
  predictions_s = []
90
  with tqdm(total=len(image_names), desc="Relocalizing") as pbar:
91
- for img_split in images.split(1, dim=0):
92
- pbar.update(1)
93
  predictions_s += to_cpu(
94
  model.reloc(img_split, ret_img=True, memory_save=False)
95
  )
 
88
  # relocalization on all images
89
  predictions_s = []
90
  with tqdm(total=len(image_names), desc="Relocalizing") as pbar:
91
+ for img_split in images.split(15, dim=0):
92
+ pbar.update(15)
93
  predictions_s += to_cpu(
94
  model.reloc(img_split, ret_img=True, memory_save=False)
95
  )