pracheeeeez commited on
Commit
f9bf9cd
·
verified ·
1 Parent(s): 4930a0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
  from langchain.chains import RetrievalQA
3
  from langchain_community.llms import HuggingFaceHub
4
- from langchain.vectorstores import Pinecone
5
  from langchain_community.embeddings import HuggingFaceEmbeddings
6
  import pinecone
7
  from pinecone import Pinecone, ServerlessSpec
@@ -12,7 +12,7 @@ host = "https://course-index-po5vpk1.svc.aped-4627-b74a.pinecone.io"
12
  pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
13
  # pc = PineconeClient(api_key=os.getenv("PINECONE_API_KEY"))
14
  index = pc.Index("course-index", host = host)
15
- vector_store = Pinecone(index=index, embedding = embedding_model, text_key = "desciption")
16
 
17
  llm = HuggingFaceHub(repo_id="t5-small", model_kwargs={"temperature": 0.5, "max_length": 64})
18
  retriever = vector_store.as_retriever()
 
1
  import streamlit as st
2
  from langchain.chains import RetrievalQA
3
  from langchain_community.llms import HuggingFaceHub
4
+ from langchain.vectorstores import LangChainPinecone
5
  from langchain_community.embeddings import HuggingFaceEmbeddings
6
  import pinecone
7
  from pinecone import Pinecone, ServerlessSpec
 
12
  pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
13
  # pc = PineconeClient(api_key=os.getenv("PINECONE_API_KEY"))
14
  index = pc.Index("course-index", host = host)
15
+ vector_store = LangChainPinecone(index=index, embedding = embedding_model, text_key = "desciption")
16
 
17
  llm = HuggingFaceHub(repo_id="t5-small", model_kwargs={"temperature": 0.5, "max_length": 64})
18
  retriever = vector_store.as_retriever()