Update app.py
Browse files
app.py
CHANGED
@@ -19,8 +19,8 @@ with open('chapter_titles.pkl', 'rb') as file:
|
|
19 |
|
20 |
def classify(text):
|
21 |
output = classifier(text, titles_astiko, multi_label=True)
|
22 |
-
output = output['sequence']
|
23 |
-
output = output[0:10]
|
24 |
return output
|
25 |
|
26 |
|
|
|
19 |
|
20 |
def classify(text):
|
21 |
output = classifier(text, titles_astiko, multi_label=True)
|
22 |
+
# output = output['sequence']
|
23 |
+
# output = output[0:10]
|
24 |
return output
|
25 |
|
26 |
|