Spaces:
Sleeping
Sleeping
Red-tech-hub
commited on
Commit
•
ba89104
1
Parent(s):
ce5e9df
[update] deploymeny
Browse files- requirements.txt +1 -0
- run.py +1 -1
requirements.txt
CHANGED
@@ -14,4 +14,5 @@ chromadb==0.5
|
|
14 |
starlette==0.37.2
|
15 |
typer==0.10.0
|
16 |
sentence-transformers
|
|
|
17 |
pinecone-client
|
|
|
14 |
starlette==0.37.2
|
15 |
typer==0.10.0
|
16 |
sentence-transformers
|
17 |
+
ctransformers
|
18 |
pinecone-client
|
run.py
CHANGED
@@ -3,7 +3,7 @@ from langchain_core.output_parsers import StrOutputParser
|
|
3 |
from langchain_core.runnables import RunnablePassthrough
|
4 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
5 |
from langchain_community.vectorstores import Chroma
|
6 |
-
from
|
7 |
|
8 |
import os
|
9 |
os.environ['TRANSFORMERS_CACHE'] = '/code/model/cache/'
|
|
|
3 |
from langchain_core.runnables import RunnablePassthrough
|
4 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
5 |
from langchain_community.vectorstores import Chroma
|
6 |
+
from ctransformers import AutoModelForCausalLM
|
7 |
|
8 |
import os
|
9 |
os.environ['TRANSFORMERS_CACHE'] = '/code/model/cache/'
|