update imports of visiontext package

lmb-freiburg · Mar 7, 2024 · b15c264 · b15c264
1 parent 78c2fa9
commit b15c264
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ovqa/models/lavis/img2prompt_models/img2prompt_vqa.py b/ovqa/models/lavis/img2prompt_models/img2prompt_vqa.py
@@ -21,7 +21,7 @@
 from ovqa.common.lavis.registry import registry
 from ovqa.models.lavis.base_model import BaseModel
 from ovqa.models.lavis.blip_models.blip_image_text_matching import compute_gradcam
-from visiontext.nlp import maybe_download_spacy_model
+from visiontext.nlp.spacytools import maybe_download_spacy_model
 
 open_pos = ["NOUN", "VERB", "ADJ", "ADV", "NUM"]
 

diff --git a/ovqa/textutils/cut_sentence.py b/ovqa/textutils/cut_sentence.py
@@ -1,6 +1,6 @@
 from nltk import word_tokenize
 
-from visiontext.nlp import (
+from visiontext.nlp.nltktools import (
     tokenize_sentences_and_connectors,
     tokenize_words_and_connectors,
     ensure_setup_nltk,