pracheeeeez commited on
Commit
bbca873
·
verified ·
1 Parent(s): 054e52e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ pc = pinecone.Pinecone(api_key=os.getenv("PINECODE_API_KEY"))
13
  index = pinecone.Index("course-index", host = host)
14
  vector_store = Pinecone(index=index, embedding = embedding_model, text_key = "desciption")
15
 
16
- llm = HuggingFaceHub(repo_id="distilbert-base-uncased", model_kwargs={"temperature": 0.5, "max_length": 64})
17
  retriever = vector_store.as_retriever()
18
  qa_chain = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever)
19
 
 
13
  index = pinecone.Index("course-index", host = host)
14
  vector_store = Pinecone(index=index, embedding = embedding_model, text_key = "desciption")
15
 
16
+ llm = HuggingFaceHub(repo_id="t5-small", model_kwargs={"temperature": 0.5, "max_length": 64})
17
  retriever = vector_store.as_retriever()
18
  qa_chain = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever)
19