pond918 commited on
Commit
130076a
·
1 Parent(s): 8fdb6b2

feat: embed array of text

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -34,14 +34,14 @@ i_search = gr.Interface(
34
  )
35
 
36
 
37
- def embed_text(text):
38
- arr = vdb.embed_text(text)
39
- json_data = json.dumps(arr.tolist())
40
- return json_data
41
 
42
  i_embed = gr.Interface(
43
- fn=embed_text,
44
- inputs=["text"],
45
  outputs="json"
46
  )
47
 
 
34
  )
35
 
36
 
37
+ def embed_texts(texts):
38
+ arr = [vdb.embed_text(t)[0].tolist() for t in texts]
39
+ json_arr = json.dumps(arr)
40
+ return json_arr
41
 
42
  i_embed = gr.Interface(
43
+ fn=embed_texts,
44
+ inputs='json',
45
  outputs="json"
46
  )
47