Fix argument parsing bug (#43)

text_threshold was wrongly set by args.box_threshold
2023-04-12 05:18:47 -04:00
parent 19e699c635
commit 049566bdc9
1 changed files with 1 additions and 1 deletions
--- 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)