diff --git a/machine-learning/src/main.py b/machine-learning/src/main.py
index 42de36a53e..a656860475 100644
--- a/machine-learning/src/main.py
+++ b/machine-learning/src/main.py
@@ -45,11 +45,8 @@ def run_engine(engine, path):
 
     for index, pred in enumerate(predictions):
         tags = pred['label'].split(', ')
-        if (index == 0):
-            result = tags
-        else:
-            if (pred['score'] > 0.5):
-                result = [*result, *tags]
+        if (pred['score'] > 0.9):
+            result = [*result, *tags]
 
     if (len(result) > 1):
         result = list(set(result))