ariana sutanto commited on
Commit
35c04e0
·
1 Parent(s): b60d96b
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -41,6 +41,8 @@ def get_score(abstract):
41
  logits = model(**inputs).logits
42
 
43
  predictions = F.softmax(logits, dim=1) #will get the probabilities of each label 0 and 1
 
 
44
  score = predictions[1] #get the probability of the label being 1 (patent accepted)
45
 
46
  return score
 
41
  logits = model(**inputs).logits
42
 
43
  predictions = F.softmax(logits, dim=1) #will get the probabilities of each label 0 and 1
44
+ print(predictions)
45
+ predictions = list(predictions)
46
  score = predictions[1] #get the probability of the label being 1 (patent accepted)
47
 
48
  return score