NEXAS commited on
Commit
5ab2a2f
·
verified ·
1 Parent(s): afb7a83

Update src/utils/ingest_image.py

Browse files
Files changed (1) hide show
  1. src/utils/ingest_image.py +2 -0
src/utils/ingest_image.py CHANGED
@@ -1,5 +1,7 @@
1
  import os
2
  import fitz
 
 
3
  import chromadb
4
  from chromadb.utils.data_loaders import ImageLoader
5
  from chromadb.utils.embedding_functions import OpenCLIPEmbeddingFunction # type: ignore
 
1
  import os
2
  import fitz
3
+ import nltk
4
+ nltk.download('punkt')
5
  import chromadb
6
  from chromadb.utils.data_loaders import ImageLoader
7
  from chromadb.utils.embedding_functions import OpenCLIPEmbeddingFunction # type: ignore