| from sentence_transformers import SentenceTransformer | |
| import gradio as gr | |
| model = SentenceTransformer('sentence-transformers/multi-qa-mpnet-base-cos-v1') | |
| def generate_embedding(query): | |
| return str(model.encode(query)) | |
| demo = gr.Interface(fn=generate_embedding, inputs="text", outputs="text") | |
| demo.launch( show_api=True) |