Changed probability calculation by removing the extra softmax function

app.py

@@ -16,8 +16,8 @@ def predictScore(abstract):
     outputs = model(inputs['input_ids'], attention_mask=inputs['attention_mask'])
     scores = torch.softmax(outputs.logits, dim=1)[0]
 
-    probs = F.softmax(scores, dim=0)
-    accept_prob = probs[0].item()
+    print(f"scores:{scores}")
+    accept_prob = scores[0].item()
     print(accept_prob)
 
     return accept_prob