Updated

app.py

@@ -14,7 +14,7 @@ model.compile(optimizer=keras.optimizers.Adam(0.001),
               loss=keras.losses.SparseCategoricalCrossentropy(from_logits=True),
               metrics=[keras.metrics.SparseCategoricalAccuracy()])
 
-model.load_weights('./weights/weights')
+model.load_weights('./weights/mnist.weights.h5')
 
 def classify(input):
     image = np.expand_dims(np.array(Image.fromarray(input['layers'][0]).resize((28,28), resample=Image.Resampling.BILINEAR), dtype=int), axis=0)

training.py

@@ -18,4 +18,4 @@ model.compile(optimizer=keras.optimizers.Adam(0.001),
 
 model.fit(x_train, y_train, validation_data=(x_test, y_test), epochs=6) 
 
-model.save_weights('./weights/weights')
+model.save_weights('./weights/mnist.weights.h5')

weights/checkpoint

@@ -1,2 +0,0 @@
-model_checkpoint_path: "weights"
-all_model_checkpoint_paths: "weights"

weights/{weights.data-00000-of-00001 → mnist.weights.h5}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:35d03d9a94c8f59a1d886350b7ce6c1170c7c3122f271ce904c1e423ec6733ab
-size 8039013
+oid sha256:e8ee2458b193128da231096cae9d7dd9c6bb3fc92a220ba992b9399a7038d7c6
+size 8062528