message

app.py

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