Update app.py

app.py

@@ -40,7 +40,7 @@ def main():
                                                "Commute Distance": [dist_dict[Commute_Distance]],
                                                "Age": [Age]}))
             # Show prediction
-            if result == 1:
+            if result == [1]:
                 i = 'Yes'
             else :
                 i = 'No'