afriddev commited on
Commit
08ef709
·
verified ·
1 Parent(s): 3bf3346

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,