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 |
|