ishaan-mital commited on
Commit
46737c8
·
1 Parent(s): 2d4bffb

initial commit

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -2,7 +2,6 @@ from gradio_client import Client
2
  import gradio as gr
3
  import requests
4
  from langchain.chains import RetrievalQA
5
- from langchain.vectorstores import Pinecone
6
  import pinecone
7
  from langchain.vectorstores import Pinecone
8
  import os
@@ -15,6 +14,12 @@ embed_model_id = 'sentence-transformers/all-MiniLM-L6-v2'
15
  embed_model = HuggingFaceEmbeddings(
16
  model_name=embed_model_id,
17
  )
 
 
 
 
 
 
18
  pinecone.init(
19
  api_key=os.environ.get('PINECONE_API_KEY'),
20
  environment=os.environ.get('PINECONE_ENVIRONMENT')
 
2
  import gradio as gr
3
  import requests
4
  from langchain.chains import RetrievalQA
 
5
  import pinecone
6
  from langchain.vectorstores import Pinecone
7
  import os
 
14
  embed_model = HuggingFaceEmbeddings(
15
  model_name=embed_model_id,
16
  )
17
+ docs = [
18
+ "this is one document",
19
+ "and another document"
20
+ ]
21
+
22
+ embeddings = embed_model.embed_documents(docs)
23
  pinecone.init(
24
  api_key=os.environ.get('PINECONE_API_KEY'),
25
  environment=os.environ.get('PINECONE_ENVIRONMENT')