cools commited on
Commit
08412aa
·
1 Parent(s): 3fef0e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,6 @@ pinecone_api_key = os.getenv("pinecone_api_key")
13
  pinecone_environment = os.getenv("pinecone_environment")
14
 
15
  s3 = boto3.client('s3', aws_access_key_id=aws_access_key, aws_secret_access_key=aws_secret_key)
16
- pinecone.init(api_key=pinecone_api_key, environment=pinecone_environment)
17
 
18
  model = SentenceTransformer('all-mpnet-base-v2')
19
 
@@ -29,7 +28,9 @@ st.divider()
29
 
30
 
31
  def search_index(query):
 
32
  index = pinecone.Index("scotus")
 
33
  vector = model.encode([query]).tolist()
34
  return index.query(vector=vector,top_k=10,include_values=False,include_metadata=True).tolist()
35
 
 
13
  pinecone_environment = os.getenv("pinecone_environment")
14
 
15
  s3 = boto3.client('s3', aws_access_key_id=aws_access_key, aws_secret_access_key=aws_secret_key)
 
16
 
17
  model = SentenceTransformer('all-mpnet-base-v2')
18
 
 
28
 
29
 
30
  def search_index(query):
31
+ pinecone.init(api_key=pinecone_api_key, environment=pinecone_environment)
32
  index = pinecone.Index("scotus")
33
+ print(index)
34
  vector = model.encode([query]).tolist()
35
  return index.query(vector=vector,top_k=10,include_values=False,include_metadata=True).tolist()
36