summerstay/vectorAPI
2
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)