hengli commited on
Commit
a8f42dc
·
1 Parent(s): 824e0dc

update hf link

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -141,6 +141,8 @@ def run_model(target_dir, model, anchor_size=100) -> dict:
141
 
142
  # Clean up
143
  torch.cuda.empty_cache()
 
 
144
  return predictions
145
 
146
 
 
141
 
142
  # Clean up
143
  torch.cuda.empty_cache()
144
+ for k,v in predictions.items():
145
+ print(f"{k}: {v.shape if isinstance(v, np.ndarray) else type(v)}")
146
  return predictions
147
 
148