Spaces:
Runtime error
Runtime error
ariana sutanto
commited on
Commit
•
e4ecba7
1
Parent(s):
35c04e0
message
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def get_score(abstract):
|
|
42 |
|
43 |
predictions = F.softmax(logits, dim=1) #will get the probabilities of each label 0 and 1
|
44 |
print(predictions)
|
45 |
-
predictions =
|
46 |
score = predictions[1] #get the probability of the label being 1 (patent accepted)
|
47 |
|
48 |
return score
|
|
|
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 |
score = predictions[1] #get the probability of the label being 1 (patent accepted)
|
47 |
|
48 |
return score
|