Pavan178 commited on
Commit
4c9b24b
·
verified ·
1 Parent(s): 327ecf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -23,7 +23,7 @@ import accelerate
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", \
@@ -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