Update app.py

app.py

@@ -51,7 +51,7 @@ def generate_images(digit):
 iface = gr.Interface(
     fn=generate_images,
     inputs=gr.Dropdown(choices=[str(i) for i in range(10)], label="Choose a digit (ignored for now)"),
-    outputs=[gr.Image(shape=(28,28), image_mode='L') for _ in range(5)],
+    outputs=[gr.Image(image_mode='L') for _ in range(5)],
     title="Handwritten Digit Generator",
     description="Select a digit (0–9) and generate 5 handwritten-style digits using a VAE trained on MNIST."
 )