Update app.py

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
 import numpy as np
-from PIL import Image
-from matplotlib import cm
 from keras.models import load_model
 
 model = load_model("QuickDraw_model.h5")
@@ -12,5 +10,15 @@ def QuickDraw(arr):
     labels = ["Apple","Cell Phone","Chair","Hot Air Balloon","Jail","Ladder","Line","Spider","Windmill","Zigzag"]
     return {labels[idx]: float(val) for idx, val in enumerate(output[0])}
 
-iface = gr.Interface(fn=QuickDraw, inputs=gr.Sketchpad().style(height=400, width=400), outputs="label")
+iface = gr.Interface(fn=QuickDraw, 
+                     inputs=gr.Image(source="canvas",
+                                    tool="sketch",
+                                    brush_radius=0.7,
+                                    invert_colors=True,
+                                    heigt=400,
+                                    width=400,
+                                    shape=(28,28),
+                                    image_mode="L",
+                                    interactive=True),
+                    outputs="label")
 iface.launch()