bluuebunny commited on
Commit
abc0ea3
·
verified ·
1 Parent(s): 0ee2db9

Update update_embeddings.py

Browse files
Files changed (1) hide show
  1. 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',