Spaces:
Runtime error
Runtime error
Commit
·
c353b99
1
Parent(s):
c0b586e
Update app.py
Browse files
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(),
|