Update app.py

app.py

@@ -96,7 +96,7 @@ optimizer = SGD(learning_rate=0.001, momentum=0.30)
 
 model.compile(optimizer = optimizer , loss = "categorical_crossentropy", metrics=["accuracy"])
 
-history = model.fit( X_train,Y_train, batch_size=64, epochs = 1, validation_data = (X_val, Y_val), verbose = 1)
+history = model.fit( X_train,Y_train, batch_size=64, epochs = 10, validation_data = (X_val, Y_val), verbose = 1)
 
 def predict_image(img):
   img_3d = img.reshape(-1, 28,28)