orhir commited on
Commit
225166a
·
verified ·
1 Parent(s): 260d09d

Update gradio_utils/utils.py

Browse files
Files changed (1) hide show
  1. gradio_utils/utils.py +1 -1
gradio_utils/utils.py CHANGED
@@ -192,7 +192,7 @@ def process(query_img, state,
192
  wrap_fp16_model(model)
193
  load_checkpoint(model, checkpoint_path, map_location='cpu')
194
  model.eval()
195
- outputs = estimate(model, str_data)
196
 
197
  # visualize results
198
  vis_s_weight = target_weight_s[0]
 
192
  wrap_fp16_model(model)
193
  load_checkpoint(model, checkpoint_path, map_location='cpu')
194
  model.eval()
195
+ outputs = estimate(model, data)
196
 
197
  # visualize results
198
  vis_s_weight = target_weight_s[0]