Update app.py

app.py

@@ -19,14 +19,17 @@ def predict(text):
     query = ' '.join(re.sub('[^가-힣a-zA-Z0-9 ]', ' ', query).split())
     result = cate_classifier(text)[0]
 
-    res = {}
+    ress = {}
+    ch = 0
+    chch = 0
     for i in range(len(result)):
-        if result[i]['score'] >= 0.5:
-            label = result[i]['label']
-            # res[f'{label}'] = 1
-            res[f'{label}'] = int(result[i]['score'] * 10000) / 100
+        if result[i]['score'] >= ch:
+            ch = result[i]['score']
+            chch = i
     
-    return res
+    ress[f'{result[chch]["label"]}'] = int(result[chch]["score"] * 10000) / 100
+    
+    return ress
 
 gr.Interface(
     predict,