Spaces:

sotosbarl
/

zero-shot

Sleeping

sotosbarl commited on Jan 6, 2024

Commit

e0b06c5

1 Parent(s): 640626f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ with open('chapter_titles.pkl', 'rb') as file:
 def classify(text):
     output = classifier(text, titles_astiko, multi_label=True)
-    output = output[0]['sequence']
     output = output[0:10]
     return output

 def classify(text):
     output = classifier(text, titles_astiko, multi_label=True)
+    output = output['sequence']
     output = output[0:10]
     return output