Spaces:
Sleeping
Sleeping
File size: 542 Bytes
3bf3346 08ef709 3bf3346 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import gradio as gr
from sentence_transformers import CrossEncoder
# Load CrossEncoder model
model = CrossEncoder("cross-encoder/nli-deberta-v3-base")
def predict_similarity(sentence1, sentence2):
score = model.predict([(sentence1, sentence2)])
return float(score[0]) # FIX: extract single float
demo = gr.Interface(
fn=predict_similarity,
inputs=["text", "text"],
outputs="number",
title="CrossEncoder (nli-deberta-v3-base)",
description="Enter two sentences to compute semantic similarity."
)
demo.launch()
|