File size: 603 Bytes
7867778 66ec8de 7867778 cf6b570 66ec8de 7867778 66ec8de cf6b570 66ec8de 7867778 66ec8de |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from sentence_transformers import SentenceTransformer
import gradio as gr
# Load a clinical sentence transformer
model = SentenceTransformer("pritamdeka/BioBERT-mnli-snli-scinli-scitail-mednli-stsb")
def embed_text(text):
embedding = model.encode(text, convert_to_numpy=True, normalize_embeddings=True).tolist()
return embedding
iface = gr.Interface(
fn=embed_text,
inputs=gr.Textbox(lines=5, label="Enter clinical text"),
outputs="json",
title="High-Accuracy Clinical Embeddings",
description="BioBERT fine-tuned for semantic similarity (STSB, MedNLI)"
)
iface.launch()
|