eval empty token_spans properly (#191)

2023-11-13 14:53:12 +09:00
parent eeba084341
commit fbb2532bb0
1 changed files with 1 additions and 1 deletions
--- a/demo/inference_on_a_image.py
+++ b/demo/inference_on_a_image.py
@@ -199,7 +199,7 @@ if __name__ == "__main__":

    # run model
    boxes_filt, pred_phrases = get_grounding_output(
-        model, image, text_prompt, box_threshold, text_threshold, cpu_only=args.cpu_only, token_spans=eval(token_spans)
+        model, image, text_prompt, box_threshold, text_threshold, cpu_only=args.cpu_only, token_spans=eval(f"{token_spans}")
    )

    # visualize pred