Update app.py
Browse files
app.py
CHANGED
|
@@ -71,8 +71,13 @@ def load_model():
|
|
| 71 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1024, chunk_overlap=256)
|
| 72 |
texts = text_splitter.split_documents(docs)
|
| 73 |
|
|
|
|
|
|
|
| 74 |
db = Chroma.from_documents(texts, embeddings, persist_directory="/home/user/app/db")
|
| 75 |
|
|
|
|
|
|
|
|
|
|
| 76 |
model_name_or_path = "TheBloke/Llama-2-13B-chat-GPTQ"
|
| 77 |
model_basename = "model"
|
| 78 |
|
|
|
|
| 71 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1024, chunk_overlap=256)
|
| 72 |
texts = text_splitter.split_documents(docs)
|
| 73 |
|
| 74 |
+
print('embedding done')
|
| 75 |
+
|
| 76 |
db = Chroma.from_documents(texts, embeddings, persist_directory="/home/user/app/db")
|
| 77 |
|
| 78 |
+
print('db done')
|
| 79 |
+
|
| 80 |
+
|
| 81 |
model_name_or_path = "TheBloke/Llama-2-13B-chat-GPTQ"
|
| 82 |
model_basename = "model"
|
| 83 |
|