Update app.py

app.py

@@ -1,5 +1,7 @@
 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")
@@ -10,5 +12,5 @@ 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="sketchpad", outputs="label")
+iface = gr.Interface(fn=QuickDraw, inputs=gr.Sketchpad().style(height=400, width=400), outputs="label")
 iface.launch()