Update app.py

app.py

@@ -76,7 +76,7 @@ data = np.array(data)
 data.shape
 
 result = np.array(result)
-result = result.reshape(139,2)
+result = result.reshape(-1,128,128,3)
 
 from sklearn.model_selection import train_test_split
 x_train,x_test,y_train,y_test = train_test_split(data, result, test_size=0.1, shuffle=True, random_state=0)