Update app.py

app.py

@@ -20,7 +20,7 @@ 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)]
+mnist_examples = [[cv2.resize(example[0], DIMS)] for example in mnist_examples]
 
 # Function to preprocess the image
 def preprocess_image(image):