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