Update app.py

app.py

@@ -0,0 +1,14 @@
+import gradio as gr
+import numpy as np
+from keras.models import load_model
+
+model = load_model("QuickDraw_model.h5")
+
+def QuickDraw(arr):
+    arr = arr.reshape((1,28,28,1))
+    output = model(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.launch()