Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,12 +8,12 @@ from langchain.chains.question_answering import load_qa_chain
|
|
| 8 |
from langchain.callbacks import get_openai_callback
|
| 9 |
from langchain import HuggingFaceHub, LLMChain
|
| 10 |
from langchain.embeddings import HuggingFaceHubEmbeddings,HuggingFaceInferenceAPIEmbeddings
|
| 11 |
-
|
| 12 |
repo_id = "sentence-transformers/all-mpnet-base-v2"
|
| 13 |
hf = HuggingFaceHubEmbeddings(
|
| 14 |
repo_id=repo_id,
|
| 15 |
task="feature-extraction",
|
| 16 |
-
|
| 17 |
)
|
| 18 |
|
| 19 |
from langchain.embeddings import HuggingFaceInferenceAPIEmbeddings
|
|
@@ -63,8 +63,8 @@ def main():
|
|
| 63 |
|
| 64 |
hub_llm = HuggingFaceHub(
|
| 65 |
repo_id='HuggingFaceH4/zephyr-7b-beta',
|
| 66 |
-
model_kwargs={'temperature':0.01,"max_length": 2048,}
|
| 67 |
-
|
| 68 |
llm = hub_llm
|
| 69 |
chain = load_qa_chain(llm, chain_type="stuff")
|
| 70 |
with get_openai_callback() as cb:
|
|
|
|
| 8 |
from langchain.callbacks import get_openai_callback
|
| 9 |
from langchain import HuggingFaceHub, LLMChain
|
| 10 |
from langchain.embeddings import HuggingFaceHubEmbeddings,HuggingFaceInferenceAPIEmbeddings
|
| 11 |
+
token = "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"
|
| 12 |
repo_id = "sentence-transformers/all-mpnet-base-v2"
|
| 13 |
hf = HuggingFaceHubEmbeddings(
|
| 14 |
repo_id=repo_id,
|
| 15 |
task="feature-extraction",
|
| 16 |
+
huggingfacehub_api_token= token,
|
| 17 |
)
|
| 18 |
|
| 19 |
from langchain.embeddings import HuggingFaceInferenceAPIEmbeddings
|
|
|
|
| 63 |
|
| 64 |
hub_llm = HuggingFaceHub(
|
| 65 |
repo_id='HuggingFaceH4/zephyr-7b-beta',
|
| 66 |
+
model_kwargs={'temperature':0.01,"max_length": 2048,},
|
| 67 |
+
huggingfacehub_api_token=token)
|
| 68 |
llm = hub_llm
|
| 69 |
chain = load_qa_chain(llm, chain_type="stuff")
|
| 70 |
with get_openai_callback() as cb:
|