File size: 585 Bytes
66ec8de 232a9f4 66ec8de 232a9f4 66ec8de 232a9f4 66ec8de 232a9f4 66ec8de |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import gradio as gr
from sentence_transformers import SentenceTransformer
# ✅ Load a model that outputs 1024-dim vectors
model = SentenceTransformer('intfloat/e5-large')
def embed_text(text):
# Optionally prepend "passage: " if using e5 models
processed_text = "passage: " + text.strip()
embedding = model.encode(processed_text).tolist()
return embedding
# Gradio interface
iface = gr.Interface(
fn=embed_text,
inputs=gr.Textbox(lines=5, label="Enter patient text"),
outputs="json",
title="Clinical Text Embedding API (1024-dim)"
)
iface.launch()
|