Update app.py

app.py

@@ -17,10 +17,18 @@ def main():
         if clicked:
             result=model.predict(pd.DataFrame({"room": [room],
                                                "partitions": [partition],
-                                               "Rows": [row],
+                                               "Rows": [Row],
                                                "Columns": [Columns],
                                                "Color": [Color]}))
-            result = 'Red' if result[0] == 1 else 'Blue'
+            if 'Color' = 'Red' : 
+                print('Red')
+            elif 'Color' = 'Blue':
+                print('BLue')
+            elif 'Color' = 'Yellow':
+                print('Yellow')
+            else 'Color' = 'Green':
+                print('Green')
+            #result = 'Red' = Red if result[0] == 1 else 'Blue'
             st.success('The predicted color is {}'.format(result))
 
 if __name__=='__main__':