vishwask commited on
Commit
9c740c6
·
1 Parent(s): 87e4cd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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