update app.py
Browse files
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
|
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, "
|
293 |
-
time.sleep(
|
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 |
|