Spaces:

1NEYRON1
/

Topic_classification_for_scientific_articles

Running

1NEYRON1 commited on Apr 4

Commit

e96ee03

1 Parent(s): 275db5d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -178,7 +178,7 @@ def classify_text(title, description, show_all=False, threshold=0.95):
         return []
     for i in results[0]:
-        i['label'] = id_to_category[int(i['label'].split('_')[1])]
     if show_all:
         filtered_results = []

         return []
     for i in results[0]:
+        i['label'] = id_to_cat[int(i['label'].split('_')[1])]
     if show_all:
         filtered_results = []