salemamassi commited on
Commit
c353b99
·
1 Parent(s): c0b586e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,7 +9,6 @@ from langchain.vectorstores import FAISS
9
  from langchain.embeddings import HuggingFaceEmbeddings
10
  from langchain.text_splitter import CharacterTextSplitter
11
  from langchain import HuggingFaceHub
12
- import time
13
 
14
  # Set your API keys
15
  API_KEY = os.environ["HUGGINGFACEHUB_API_TOKEN"]
@@ -38,7 +37,7 @@ def chat(message,history):
38
  history_langchain_format.append(HumanMessage(content=message))
39
  history_langchain_format.append(HumanMessage(content=message))
40
  # Create the index (update index)
41
- llm2 = HuggingFaceHub(repo_id="declare-lab/flan-alpaca-large", model_kwargs={"temperature": 0, "max_length": 512},API_KEY )
42
  chain = RetrievalQA.from_chain_type(llm=llm2,
43
  chain_type="stuff",
44
  retriever=index.vectorstore.as_retriever(),
 
9
  from langchain.embeddings import HuggingFaceEmbeddings
10
  from langchain.text_splitter import CharacterTextSplitter
11
  from langchain import HuggingFaceHub
 
12
 
13
  # Set your API keys
14
  API_KEY = os.environ["HUGGINGFACEHUB_API_TOKEN"]
 
37
  history_langchain_format.append(HumanMessage(content=message))
38
  history_langchain_format.append(HumanMessage(content=message))
39
  # Create the index (update index)
40
+ llm2 = HuggingFaceHub(repo_id="declare-lab/flan-alpaca-large", model_kwargs={"temperature": 0, "max_length": 512},huggingfacehub_api_token = API_KEY )
41
  chain = RetrievalQA.from_chain_type(llm=llm2,
42
  chain_type="stuff",
43
  retriever=index.vectorstore.as_retriever(),