orhir commited on
Commit
e6f223b
·
verified ·
1 Parent(s): b61e4ea

Update gradio_utils/utils.py

Browse files
Files changed (1) hide show
  1. gradio_utils/utils.py +1 -1
gradio_utils/utils.py CHANGED
@@ -194,7 +194,7 @@ def process(query_img, state,
194
  load_checkpoint(model, checkpoint_path, map_location='cpu')
195
  model.eval()
196
  outputs = estimate(model, data)
197
- print("Output Avail!")
198
  # visualize results
199
  vis_s_weight = target_weight_s[0]
200
  vis_s_image = support_img[0].detach().cpu().numpy().transpose(1, 2, 0)
 
194
  load_checkpoint(model, checkpoint_path, map_location='cpu')
195
  model.eval()
196
  outputs = estimate(model, data)
197
+ print("Output Avail!")
198
  # visualize results
199
  vis_s_weight = target_weight_s[0]
200
  vis_s_image = support_img[0].detach().cpu().numpy().transpose(1, 2, 0)