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", \
|