Update app.py

app.py

@@ -47,9 +47,18 @@ def recognize_digit(image):
         return {str(i): float(prediction[0][i]) for i in range(10)}
     else:
         return ""
-    
+
+def create_canvas():
+    fig, ax = plt.subplots()
+    ax.set_title("Draw your digit")
+    ax.set_xticks([])
+    ax.set_yticks([])
+    canvas = FigureCanvas(fig)
+    return canvas
+
+
 
 demo = gr.Interface(fn=recognize_digit, 
-                    inputs=gr.Image(image_mode="L"), 
+                    inputs=gr.Image(image_mode="L", source="canvas", canvas=canvas), 
                     outputs=gr.Label(num_top_classes=1))
 demo.launch(share=True)