Add debug prints for predictions and class names; change ImageEditor to Sketchpad

app.py

@@ -34,10 +34,13 @@ def predict(data):
     # Model predictions
     preds = model.predict(img)[0]
 
+    print(preds)
+
     # Get top 3 classes
     top_3_classes = np.argsort(preds)[-3:][::-1]
     top_3_probs = preds[top_3_classes]
     class_names = [labels[i] for i in top_3_classes]
+    print(class_names, top_3_probs, top_3_classes)
 
     # Return top 3 predictions as a dictionary
     return {class_names[i]: float(top_3_probs[i]) for i in range(3)}
@@ -60,12 +63,12 @@ with gr.Blocks(title=title) as demo:
 
     with gr.Row():
         # Using ImageEditor with type='numpy'
-        im = gr.ImageEditor(type="numpy", label="Draw your digit here (use brush and eraser)")
+        im = gr.Sketchpad(type="numpy", label="Draw your digit here (use brush and eraser)")
 
         # Output label (top 3 predictions)
         label = gr.Label(num_top_classes=3, label="Predictions")
 
     # Trigger prediction whenever the image changes
-    im.change(predict, inputs=im, outputs=label, show_progress="hidden")
+    im.change(predict, inputs=im, outputs=label)
 
     demo.launch(share=True)