message

app.py

@@ -44,7 +44,7 @@ def get_score(abstract):
     print(predictions)
     predictions = predictions.numpy()
     print(predictions)
-    score = predictions[0] #get the probability of the label being 1 (patent accepted)
+    score = predictions[0][1] #get the probability of the label being 1 (patent accepted)
 
     return score