From fbb2532bb05e329c32533cf3b65aa360b574051e Mon Sep 17 00:00:00 2001 From: Kazuto Murase Date: Mon, 13 Nov 2023 14:53:12 +0900 Subject: [PATCH] eval empty token_spans properly (#191) --- demo/inference_on_a_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/inference_on_a_image.py b/demo/inference_on_a_image.py index 0dd332f..63f09bd 100644 --- 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