Update app.py

app.py

@@ -24,8 +24,14 @@ def main():
                                                "Rows": [Rows],
                                                "Columns": [Columns],
                                                "Color": [Color]}))
-        
-            if result == 'Red':, elif result == 'Blue':, elif result == 'Yellow':, and else:
+            if result == 'Red':
+                print('Red')
+            elif result == 'Blue':
+                print('Blue')
+            elif result == 'Yellow':
+                print('Yellow')
+            else:
+                print('Green')
             #result = 'Red' = Red if result[0] == 1 else 'Blue'
             st.success('The predicted color is {}'.format(result))