Spaces:
Runtime error
Runtime error
Hong
commited on
Commit
·
be4181e
1
Parent(s):
869230c
Update BART_utils.py
Browse files- BART_utils.py +1 -1
BART_utils.py
CHANGED
@@ -45,7 +45,7 @@ def get_prob(sequence, label):
|
|
45 |
|
46 |
def get_taggs(sequence, labels, thred=0.5):
|
47 |
out = []
|
48 |
-
for i in stqdm(range(len(labels)):
|
49 |
l = labels[i]
|
50 |
temp = get_prob(sequence, l)
|
51 |
if temp >= thred:
|
|
|
45 |
|
46 |
def get_taggs(sequence, labels, thred=0.5):
|
47 |
out = []
|
48 |
+
for i in stqdm(range(len(labels))):
|
49 |
l = labels[i]
|
50 |
temp = get_prob(sequence, l)
|
51 |
if temp >= thred:
|