| import gradio as gr | |
| from sentence_transformers import SentenceTransformer | |
| # Load the model | |
| model = SentenceTransformer('sentence-transformers/msmarco-distilbert-dot-v5') | |
| # Function to get the embedding | |
| def embedding(text): | |
| text_emb = model.encode(text) | |
| return text_emb | |
| # Define the Streamlit app | |
| gradio_app = gr.Interface( | |
| embedding, | |
| inputs=gr.Text(label="TEXT"), | |
| outputs=gr.Text(label="Embedding"), | |
| title="Embedding", | |
| ) | |
| if __name__ == "__main__": | |
| gradio_app.launch() |