Update app.py

app.py

@@ -5,8 +5,8 @@ from sentence_transformers import CrossEncoder
 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
+    score = model.predict([(sentence1, sentence2)])  # returns numpy array
+    return float(score.squeeze()[0])  # ✅ safe extraction of single float
 
 demo = gr.Interface(
     fn=predict_similarity,