Thao Pham commited on
Commit
527422b
·
1 Parent(s): d50ce1c

allow for creating index

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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 = "mutlimodal-minilm"
 
 
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