Update Home.py

Home.py

@@ -18,9 +18,9 @@ def load_essential_models(scaler_6_path, scaler_8_path, scaler_full_path, clf_6_
     return scaler_6, scaler_8, scaler_21, clf_6, clf_8, clf_21
 
 def convert_prediction(prediction):
-    if prediction == [0]:
+    if (prediction[0] == [0]):
         return 'S'
-    elif prediction == [1]:
+    elif (prediction[0] == [1]):
         return 'M'
     else:
         return 'L'