switched from using sketchpad

app.py

@@ -7,13 +7,9 @@ from PIL import Image
 model = tf.keras.models.load_model("mnist_model.h5")
 
 def predict(image):
-    # Extract numpy array if dict
-    if isinstance(image, dict) and "image" in image:
-        image = image["image"]
-
-    # Convert to PIL
-    image = Image.fromarray(image.astype("uint8")).convert("L").resize((28, 28))
-
+    # Ensure PIL image, convert to grayscale, resize
+    image = image.convert("L").resize((28, 28))
+    
     # Normalize
     image = np.array(image) / 255.0
     image = image.reshape(1, 28, 28, 1)
@@ -22,12 +18,11 @@ def predict(image):
     prediction = model.predict(image)
     return str(np.argmax(prediction))
 
-# Gradio interface
+# Gradio interface (upload instead of sketchpad)
 iface = gr.Interface(
     fn=predict,
-    inputs=gr.Sketchpad(),
-    outputs="label",
-    live=True
+    inputs=gr.Image(type="pil", image_mode="L", shape=(28,28)),
+    outputs="label"
 )
 
 iface.launch()