Update app.py

app.py

@@ -14,7 +14,11 @@ model = load_model('mnist_model.h5')
 
 # Load MNIST examples
 (x_train, y_train), (x_test, y_test) = mnist.load_data()
-mnist_examples = [[x_test[i]] for i in range(10)]  # Select first 10 examples and format as nested list
+# Select one example for each digit 0-9
+mnist_examples = []
+for digit in range(10):
+    idx = np.where(y_test == digit)[0][0]
+    mnist_examples.append([x_test[idx]])
 # Resize the examples to 100 by 100
 mnist_examples = [[cv2.resize(x_test[i], DIMS)] for i in range(10)]