Update app.py

app.py

@@ -31,4 +31,16 @@ def predict(im):
         out = model(x)
     probabilities = torch.nn.functional.softmax(out[0], dim=0)
     values, indices = torch.topk(probabilities, 5)
-    return {LABELS[i]: v.item() for i, v in zip(indices, values)}
+    return {LABELS[i]: v.item() for i, v in zip(indices, values)}
+
+interface = gr.Interface(
+    predict,
+    inputs="sketchpad",
+    outputs="label",
+    theme="huggingface",
+    title="Sketch Recognition",
+    description="Who wants to play Pictionary? Draw a common object like a shovel or a laptop, and the algorithm will guess in real time!",
+    article="<p style='text-align: center'>Sketch Recognition | Demo Model</p>",
+    live=True,
+)
+interface.launch(share=True)