AlekseyV commited on
Commit
05faee0
·
verified ·
1 Parent(s): 6a5d55a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,14 +4,14 @@ from fastembed import SparseTextEmbedding
4
  def sparseembed(docs):
5
  model = SparseTextEmbedding(model_name="Qdrant/bm25")
6
  embeddings = list(model.embed(docs))
7
- return [(x.values, x.indicies) for x in embeddings]
8
 
9
  iface = gr.Interface(
10
  fn=sparseembed,
11
  inputs=[
12
  gr.JSON(label="Docs (JSON array of objects)")
13
  ],
14
- outputs=gr.Dataframe(type="array", headers=["values", "indicies"]),
15
  api_name="rerank"
16
  )
17
 
 
4
  def sparseembed(docs):
5
  model = SparseTextEmbedding(model_name="Qdrant/bm25")
6
  embeddings = list(model.embed(docs))
7
+ return [(x.values, x.indices) for x in embeddings]
8
 
9
  iface = gr.Interface(
10
  fn=sparseembed,
11
  inputs=[
12
  gr.JSON(label="Docs (JSON array of objects)")
13
  ],
14
+ outputs=gr.Dataframe(type="array", headers=["values", "indices"]),
15
  api_name="rerank"
16
  )
17