Spaces:

djsull
/

aha-multi-label

Runtime error

djsull commited on Sep 28, 2022

Commit

ad29555

•

1 Parent(s): 72d67dc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,8 +26,8 @@ def predict(text):
         if result[i]['score'] >= ch:
             ch = result[i]['score']
             chch = i
-    ress[f'{result[chch]["label"]}'] = int(result[chch]["score"] * 10000) / 100
     return ress

         if result[i]['score'] >= ch:
             ch = result[i]['score']
             chch = i
+    text_tmp = result[chch]["label"]
+    ress[text_tmp] = int(result[chch]["score"] * 10000) / 100
     return ress