Refactor predict function: add debug prints for non-zero pixel values and update input sketchpad to grayscale mode

app.py

@@ -39,6 +39,14 @@ def predict(data):
     # Convert to NumPy array
     img = np.array(data['composite'])
 
+    # print non-zero values
+    print("non-zero values", np.count_nonzero(img))
+    for i in range(img.shape[0]):
+        for j in range(img.shape[1]):
+            for k in range(img.shape[2]):
+                if img[i][j][k] != 0:
+                    print("img[i][j][k]", img[i][j][k])
+
     print("img.shape", img.shape)
 
     # Handle RGBA or RGB images
@@ -80,7 +88,7 @@ label = gr.Label(num_top_classes=3)
 # Open Gradio interface for sketch recognition
 interface = gr.Interface(
     fn=predict,
-    inputs=gr.Sketchpad(type='numpy'),
+    inputs=gr.Sketchpad(type='numpy', image_mode='L', brush=gr.Brush()),
     outputs=label,
     title=title,
     description=head,