Update update_embeddings.py
Browse files- update_embeddings.py +3 -1
update_embeddings.py
CHANGED
@@ -20,7 +20,7 @@ import numpy as np # For array manipulation
|
|
20 |
from huggingface_hub import HfApi # To transact with huggingface.co
|
21 |
import sys # To quit the script
|
22 |
import datetime # get current year
|
23 |
-
from time import time # To time the script
|
24 |
|
25 |
# Start timer
|
26 |
start = time()
|
@@ -150,6 +150,8 @@ def embed(input_text):
|
|
150 |
|
151 |
else:
|
152 |
|
|
|
|
|
153 |
# Calculate embeddings by calling mxbai.embeddings()
|
154 |
result = mxbai.embeddings(
|
155 |
model='mixedbread-ai/mxbai-embed-large-v1',
|
|
|
20 |
from huggingface_hub import HfApi # To transact with huggingface.co
|
21 |
import sys # To quit the script
|
22 |
import datetime # get current year
|
23 |
+
from time import time, sleep # To time the script
|
24 |
|
25 |
# Start timer
|
26 |
start = time()
|
|
|
150 |
|
151 |
else:
|
152 |
|
153 |
+
sleep(0.2)
|
154 |
+
|
155 |
# Calculate embeddings by calling mxbai.embeddings()
|
156 |
result = mxbai.embeddings(
|
157 |
model='mixedbread-ai/mxbai-embed-large-v1',
|