Update app.py

app.py

@@ -10,7 +10,7 @@ def classify_digit(image):
     if image is None: 
         return None
     
-    # Robust check: Gradio 4.x Sketchpad might return a dictionary
+    # Robust check: Gradio 4.x Sketchpad returns a dictionary {'background':..., 'layers':..., 'composite':...}
     if isinstance(image, dict):
         image = image['composite']
         
@@ -46,7 +46,6 @@ def classify_digit(image):
     return {str(i): float(prediction[i]) for i in range(10)}
 
 # --- UI SETUP ---
-# We use gr.Blocks to create a custom layout with Tabs
 with gr.Blocks() as demo:
     gr.Markdown("## Handwritten Digit Recognizer")
     gr.Markdown("Draw a digit (0-9) or upload a photo to test the model.")
@@ -54,12 +53,17 @@ with gr.Blocks() as demo:
     with gr.Tabs():
         # Tab 1: Drawing Interface
         with gr.Tab("Draw Digit"):
-            sketchpad = gr.Sketchpad(label="Draw Here", type="numpy", brush=gr.Brush(color="#000000", thickness=20))
+            # FIX: Use 'default_size' and 'colors' instead of 'thickness' and 'color'
+            sketchpad = gr.Sketchpad(
+                label="Draw Here", 
+                type="numpy", 
+                brush=gr.Brush(colors=["#000000"], default_size=20)
+            )
             btn_draw = gr.Button("Predict Drawing", variant="primary")
             
         # Tab 2: Upload Interface
         with gr.Tab("Upload Photo"):
-            # sources=["upload", "clipboard"] fixes your specific error
+            # FIX: Use 'sources=["upload", "clipboard"]' to avoid the source list error
             upload = gr.Image(label="Upload Image", sources=["upload", "clipboard"], type="numpy")
             btn_upload = gr.Button("Predict Upload", variant="primary")