Spaces:
Running
on
Zero
Running
on
Zero
Update gradio_utils/utils.py
Browse files- 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,
|
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]
|