sketch

app.py

@@ -33,7 +33,7 @@ model.eval()
 # Gradio preprocessing and prediction pipeline
 def predict_digit(image):
     # Preprocess the image: resize to 28x28, convert to grayscale, and normalize
-    image = image.convert('L')  # Convert to grayscale
+    image = Image.fromarray(image).convert('L')  # Convert to grayscale
     transform = transforms.Compose([
         transforms.Resize((28, 28)),
         transforms.ToTensor(),
@@ -52,7 +52,7 @@ def predict_digit(image):
 # Create Gradio Interface
 interface = gr.Interface(
     fn=predict_digit,
-    inputs=gr.Image(source="canvas", tool="editor", type="pil"),  # User can draw on a canvas
+    inputs=gr.Sketchpad(shape=(28, 28)),  # Sketchpad for users to draw
     outputs="text",
     title="Digit Recognizer",
     description="Draw a digit (0-9) and the model will predict the number!"