Spaces:
Sleeping
Sleeping
File size: 577 Bytes
6a5d55a 7dacf2f 6a5d55a 05faee0 6a5d55a 7dacf2f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import gradio as gr
from fastembed import SparseTextEmbedding
def sparseembed(docs):
model = SparseTextEmbedding(model_name="Qdrant/bm25")
embeddings = list(model.embed(docs))
# преобразуем x.values и x.indices в list
return [ (x.values.tolist(), x.indices.tolist()) for x in embeddings ]
iface = gr.Interface(
fn=sparseembed,
inputs=[
gr.JSON(label="Docs (JSON array of objects)")
],
outputs=gr.Dataframe(type="array", headers=["values", "indices"]),
api_name="rerank"
)
if __name__ == "__main__":
iface.launch() |