Update app.py
Browse files
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 |
|