message

app.py

@@ -43,7 +43,8 @@ def get_score(abstract):
     predictions = F.softmax(logits, dim=1) #will get the probabilities of each label 0 and 1
     print(predictions)
     predictions = predictions.numpy()
-    score = predictions[1] #get the probability of the label being 1 (patent accepted)
+    print(predictions)
+    score = predictions[0] #get the probability of the label being 1 (patent accepted)
 
     return score