Add new labels to result

app.py

@@ -9,9 +9,9 @@ from transformers import (
 )
 
 labels = [
-    'agency',
+    # 'agency',
     # 'humanComparison',
-    'hyperbole',
+    # 'hyperbole',
     # 'historyComparison',
     # 'unjustClaims',
     # 'deepSounding',
@@ -20,6 +20,24 @@ labels = [
     # 'performanceNumber',
     # 'inscrutable',
     # 'objective'
+    "agency",
+    "suggestiveImagery",
+    "comparisonWithHumanIntelligence",
+    "comparisonWithHumanSkills",
+    "hyperbole",
+    "uncriticalHistoryComparison",
+    "unjustifiedClaimsAboutFuture",
+    "falseClaimsAboutProgress",
+    "incorrectClaimsAboutStudyReport",
+    "deepSoundingTermsForBanalities",
+    "treatingSpokespeopleAsNeutral",
+    "repeatingPRTerms",
+    "noDiscussionOfLimitations",
+    "deEmphasizingLimitations",
+    "limitationsAddressedBySkeptics",
+    "downplayingHumanLabour",
+    "performanceNumbersWithoutCaveats",
+    "inscrutability",
 ]
 
 models = {}
@@ -49,6 +67,7 @@ def predict(text):
     return preds
 
 
+d
 examples = [
     "Machine Learning is at the forefront of education, replacing human jobs",
     "AI model leaves scientists confused",