Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ model = CrossEncoder("cross-encoder/nli-deberta-v3-base")
|
|
| 6 |
|
| 7 |
def predict_similarity(sentence1, sentence2):
|
| 8 |
score = model.predict([(sentence1, sentence2)])
|
| 9 |
-
return float(score)
|
| 10 |
|
| 11 |
demo = gr.Interface(
|
| 12 |
fn=predict_similarity,
|
|
|
|
| 6 |
|
| 7 |
def predict_similarity(sentence1, sentence2):
|
| 8 |
score = model.predict([(sentence1, sentence2)])
|
| 9 |
+
return float(score[0]) # FIX: extract single float
|
| 10 |
|
| 11 |
demo = gr.Interface(
|
| 12 |
fn=predict_similarity,
|