Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ from langchain_community.llms import HuggingFacePipeline
|
|
10 |
from langchain.chains import ConversationChain
|
11 |
from langchain.memory import ConversationBufferMemory
|
12 |
from langchain_community.llms import HuggingFaceEndpoint
|
13 |
-
|
14 |
from pathlib import Path
|
15 |
import chromadb
|
16 |
from unidecode import unidecode
|
@@ -23,7 +23,7 @@ import accelerate
|
|
23 |
import re
|
24 |
|
25 |
|
26 |
-
|
27 |
# default_persist_directory = './chroma_HF/'
|
28 |
list_llm = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.1", \
|
29 |
"google/gemma-7b-it","google/gemma-2b-it", \
|
|
|
10 |
from langchain.chains import ConversationChain
|
11 |
from langchain.memory import ConversationBufferMemory
|
12 |
from langchain_community.llms import HuggingFaceEndpoint
|
13 |
+
import spaces
|
14 |
from pathlib import Path
|
15 |
import chromadb
|
16 |
from unidecode import unidecode
|
|
|
23 |
import re
|
24 |
|
25 |
|
26 |
+
@spaces.GPU
|
27 |
# default_persist_directory = './chroma_HF/'
|
28 |
list_llm = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.1", \
|
29 |
"google/gemma-7b-it","google/gemma-2b-it", \
|