Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -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", \
|
@@ -34,6 +34,7 @@ list_llm = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mixtral-8x7B-Instru
|
|
34 |
]
|
35 |
list_llm_simple = [os.path.basename(llm) for llm in list_llm]
|
36 |
|
|
|
37 |
# Load PDF document and create doc splits
|
38 |
def load_doc(list_file_path, chunk_size, chunk_overlap):
|
39 |
# Processing for one document only
|
|
|
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", \
|
|
|
34 |
]
|
35 |
list_llm_simple = [os.path.basename(llm) for llm in list_llm]
|
36 |
|
37 |
+
@spaces.GPU
|
38 |
# Load PDF document and create doc splits
|
39 |
def load_doc(list_file_path, chunk_size, chunk_overlap):
|
40 |
# Processing for one document only
|