Spaces:
Running
Running
Update src/utils/ingest_image.py
Browse files
src/utils/ingest_image.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import os
|
2 |
import fitz
|
3 |
-
|
4 |
-
|
5 |
import chromadb
|
6 |
from chromadb.utils.data_loaders import ImageLoader
|
7 |
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
|