pracheeeeez commited on
Commit
657d09f
·
verified ·
1 Parent(s): b037f58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,10 +5,10 @@ from langchain_community.vectorstores import Pinecone
5
  import pinecone
6
  import os
7
  from sentence_transformers import SentenceTransformer
8
-
9
- pinecone.init(api_key=os.getenv("PINECONE_API_KEY"), environment="us-west1-gcp")
10
  index = pinecone.Index("course-index")
11
- vector_store = Pinecone(index=index)
12
 
13
  llm = HuggingFaceHub(repo_id="distilbert-base-uncased", model_kwargs={"temperature": 0.5, "max_length": 64})
14
  retriever = vector_store.as_retriever()
 
5
  import pinecone
6
  import os
7
  from sentence_transformers import SentenceTransformer
8
+ embedding_model = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
9
+ pc = pinecone.Pinecone(api_key=os.getenv("PINECONE_API_KEY"), environment="us-west1-gcp")
10
  index = pinecone.Index("course-index")
11
+ vector_store = Pinecone(index=index, embedding = embedding_mdeol, text_key = "desciption")
12
 
13
  llm = HuggingFaceHub(repo_id="distilbert-base-uncased", model_kwargs={"temperature": 0.5, "max_length": 64})
14
  retriever = vector_store.as_retriever()