Spaces:

stevengrove
/

YOLO-World

Runtime error

stevengrove commited on Jan 31, 2024

Commit

fd61144

verified ·

1 Parent(s): de00aaf

Update tools/demo.py

Files changed (1) hide show

tools/demo.py CHANGED Viewed

@@ -194,7 +194,7 @@ def demo(runner, args, cfg):
                      [output_image])
         clear.click(lambda: [[], '', ''], None,
                     [image, input_text, output_image])
-        export.click(partial(export_model, runner, cfg.checkpoint),
                      [input_text, max_num_boxes, score_thr, nms_thr],
                      [out_download, out_download])
         demo.launch(server_name='0.0.0.0')

                      [output_image])
         clear.click(lambda: [[], '', ''], None,
                     [image, input_text, output_image])
+        export.click(partial(export_model, runner, args.checkpoint),
                      [input_text, max_num_boxes, score_thr, nms_thr],
                      [out_download, out_download])
         demo.launch(server_name='0.0.0.0')