Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
3784e1c
1
Parent(s):
6e7cb96
fix
Browse files- utils/summary_utils.py +6 -3
utils/summary_utils.py
CHANGED
@@ -71,9 +71,12 @@ def get_tags(text: str):
|
|
71 |
print(tag_gen_pipe_1(text))
|
72 |
print(tag_gen_pipe_2(text))
|
73 |
print(tag_gen_pipe_3(text))
|
74 |
-
|
75 |
-
|
76 |
-
|
|
|
|
|
|
|
77 |
# print(tags1, tags2, tags3)
|
78 |
return []
|
79 |
except:
|
|
|
71 |
print(tag_gen_pipe_1(text))
|
72 |
print(tag_gen_pipe_2(text))
|
73 |
print(tag_gen_pipe_3(text))
|
74 |
+
tags1 = [tag['label'] for tag in tag_gen_pipe_1(text) if tag['score'] >= MinTagScore]
|
75 |
+
tags2 = [tag['label'] for tag in tag_gen_pipe_2(text) if tag['score'] >= MinTagScore]
|
76 |
+
tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
|
77 |
+
print(tags1)
|
78 |
+
print(tags2)
|
79 |
+
print(tags3)
|
80 |
# print(tags1, tags2, tags3)
|
81 |
return []
|
82 |
except:
|