Test fix for #11 (#12)

2023-03-28 03:39:44 +02:00
parent 858efccbad
commit c974f60d73
3 changed files with 8 additions and 21 deletions
--- a/groundingdino/util/inference.py
+++ b/groundingdino/util/inference.py
@@ -71,7 +71,7 @@ def predict(
    tokenized = tokenizer(caption)

    phrases = [
-        get_phrases_from_posmap(logit > text_threshold, tokenized, caption).replace('.', '')
+        get_phrases_from_posmap(logit > text_threshold, tokenized, tokenizer).replace('.', '')
        for logit
        in logits
    ]