CoolFace
Apppublic

summerstay/vectorAPI

sourceHugging Faceccupdated 4y agoView on Hugging Face
2likes
app.py20 linesDownload Raw Back to root
1 2import gradio as gr3from sentence_transformers import SentenceTransformer4 5model_Q = SentenceTransformer('sentence-transformers/all-mpnet-base-v2')6 7def getVectors(sentences):8            vectors = []9            splitSentences = sentences.split('ZZZ')10            for sentence in splitSentences:11                       vectors.append(model_Q.encode(sentence).tolist())12            return vectors13            14interface = gr.Interface(fn = getVectors,15inputs = "text",16outputs = ['text'],17title = 'get vectors',18description = 'get vectors for search')19 20interface.launch(inline = False)