Spaces:
Sleeping
Sleeping
File size: 622 Bytes
3592343 ccc69c6 3592343 ccc69c6 3943339 3592343 ccc69c6 3592343 ccc69c6 3943339 3592343 3943339 3592343 ccc69c6 3592343 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# app.py
import gradio as gr
from sentence_transformers import SentenceTransformer
# Initialize the embedding model
model = SentenceTransformer('distilbert-base-nli-stsb-mean-tokens')
def get_embedding(text):
embedding = model.encode(text)
return embedding.tolist()
# Define the Gradio interface using updated components
iface = gr.Interface(
fn=get_embedding,
inputs=gr.Textbox(lines=5, placeholder="Enter text here..."),
outputs=gr.JSON(),
title="Embedding Generation Service",
description="Generates sentence embeddings using DistilBERT."
)
if __name__ == "__main__":
iface.launch()
|