Spaces:
Runtime error
Runtime error
ariana sutanto
commited on
Commit
•
86f5bbf
1
Parent(s):
7fe46a2
message
Browse files
app.py
CHANGED
@@ -41,9 +41,7 @@ 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 |
-
print(predictions)
|
45 |
predictions = predictions.numpy()
|
46 |
-
print(predictions)
|
47 |
score = predictions[0][1] #get the probability of the label being 1 (patent accepted)
|
48 |
|
49 |
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 |
predictions = predictions.numpy()
|
|
|
45 |
score = predictions[0][1] #get the probability of the label being 1 (patent accepted)
|
46 |
|
47 |
return score
|