Manel commited on
Commit
e8a5c54
·
verified ·
1 Parent(s): 874464b

update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -57,7 +57,7 @@ def load_db(device, local_embed=False, CHROMA_PATH = './ChromaDB'):
57
  embed_id = "BAAI/bge-large-en-v1.5"
58
  start_time = time.time()
59
 
60
- #TODO : LOOK INTO LOADING ONLY A SINGLE FILE FROM HF REPO TO REDUCE MEMORY
61
  if local_embed:
62
  from transformers import AutoModel
63
 
@@ -289,8 +289,8 @@ if __name__=="__main__":
289
  db = load_db(device)
290
  bar.progress(0.7, "Loading Model. Please wait.")
291
  model, tokenizer = load_model(model_name)
292
- bar.progress(1.0, "Now you can type your question.")
293
- time.sleep(1)
294
  bar.empty()
295
 
296
 
 
57
  embed_id = "BAAI/bge-large-en-v1.5"
58
  start_time = time.time()
59
 
60
+ #TODO : LOOK INTO LOADING ONLY A SINGLE EMBEDDING FILE TO REDUCE LOADING TIME
61
  if local_embed:
62
  from transformers import AutoModel
63
 
 
289
  db = load_db(device)
290
  bar.progress(0.7, "Loading Model. Please wait.")
291
  model, tokenizer = load_model(model_name)
292
+ bar.progress(1.0, "Ready to ask? Go ahead and type your question.")
293
+ time.sleep(2)
294
  bar.empty()
295
 
296