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 |
|