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

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from fastembed import SparseTextEmbedding
3
+
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
+
18
+ if __name__ == "__main__":
19
+ iface.launch()