Thao Pham
commited on
Commit
·
527422b
1
Parent(s):
d50ce1c
allow for creating index
Browse files
app.py
CHANGED
@@ -8,6 +8,7 @@ import rag
|
|
8 |
import os
|
9 |
import uuid
|
10 |
import numpy as np
|
|
|
11 |
from pinecone import Pinecone, ServerlessSpec
|
12 |
from sentence_transformers import SentenceTransformer
|
13 |
from transformers import AutoImageProcessor, AutoModel
|
@@ -35,8 +36,11 @@ pc = Pinecone(
|
|
35 |
api_key=PINECONE_API_KEY
|
36 |
)
|
37 |
# Connect to an index
|
38 |
-
index_name = "
|
|
|
|
|
39 |
INDEX = pc.Index(index_name)
|
|
|
40 |
MODEL_STACK = [TEXT_MODEL, VISION_MODEL, VISION_MODEL_PROCESSOR, VLM, VLM_PROCESSOR]
|
41 |
|
42 |
|
|
|
8 |
import os
|
9 |
import uuid
|
10 |
import numpy as np
|
11 |
+
import pinecone
|
12 |
from pinecone import Pinecone, ServerlessSpec
|
13 |
from sentence_transformers import SentenceTransformer
|
14 |
from transformers import AutoImageProcessor, AutoModel
|
|
|
36 |
api_key=PINECONE_API_KEY
|
37 |
)
|
38 |
# Connect to an index
|
39 |
+
index_name = "multimodal-minilm"
|
40 |
+
if index_name not in pc.list_indexes().names():
|
41 |
+
pinecone.create_index(index_name, dimension=384, metric="cosine")
|
42 |
INDEX = pc.Index(index_name)
|
43 |
+
|
44 |
MODEL_STACK = [TEXT_MODEL, VISION_MODEL, VISION_MODEL_PROCESSOR, VLM, VLM_PROCESSOR]
|
45 |
|
46 |
|