amezi commited on
Commit
2db5cdf
·
1 Parent(s): fb3036a

changing pinecone dims

Browse files
Files changed (1) hide show
  1. src/pinecone_store.py +4 -4
src/pinecone_store.py CHANGED
@@ -8,15 +8,15 @@ class PineconeStore:
8
  def __init__(self):
9
  self.pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
10
 
11
- if 'blip-soccer-highlights' not in self.pc.list_indexes().names():
12
  self.pc.create_index(
13
- name='blip-soccer-highlights',
14
- dimension=1024, # This must match InternVL2_5-8B-MPO
15
  metric='cosine',
16
  spec=ServerlessSpec(cloud='aws', region=os.getenv("PINECONE_ENV", "us-east-1"))
17
  )
18
 
19
- self.index = self.pc.Index('blip-soccer-highlights')
20
 
21
  def upsert(self, id, vector, metadata):
22
  self.index.upsert([(id, vector.tolist(), metadata)])
 
8
  def __init__(self):
9
  self.pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
10
 
11
+ if 'blip-soccer-highlights-index' not in self.pc.list_indexes().names():
12
  self.pc.create_index(
13
+ name='blip-soccer-highlights-index',
14
+ dimension=768, # This must match InternVL2_5-8B-MPO
15
  metric='cosine',
16
  spec=ServerlessSpec(cloud='aws', region=os.getenv("PINECONE_ENV", "us-east-1"))
17
  )
18
 
19
+ self.index = self.pc.Index('blip-soccer-highlights-index')
20
 
21
  def upsert(self, id, vector, metadata):
22
  self.index.upsert([(id, vector.tolist(), metadata)])