milyiyo commited on
Commit
c9916d8
·
1 Parent(s): ba811c0

Debug error inside get_embeddings

Browse files
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -49,7 +49,7 @@ def get_embeddings(text):
49
  # emb_item = model_output.detach().cpu().numpy()[0]
50
  # print(emb_item)
51
  print(['get_embeddings', 'end'])
52
- return emb_item
53
 
54
 
55
  def build_faiss_index(text):
 
49
  # emb_item = model_output.detach().cpu().numpy()[0]
50
  # print(emb_item)
51
  print(['get_embeddings', 'end'])
52
+ return model_output
53
 
54
 
55
  def build_faiss_index(text):