diff --git a/demo/inference_on_a_image.py b/demo/inference_on_a_image.py index 62546d7..207227b 100644 --- a/demo/inference_on_a_image.py +++ b/demo/inference_on_a_image.py @@ -143,7 +143,7 @@ if __name__ == "__main__": text_prompt = args.text_prompt output_dir = args.output_dir box_threshold = args.box_threshold - text_threshold = args.box_threshold + text_threshold = args.text_threshold # make dir os.makedirs(output_dir, exist_ok=True)