Testing george shakan tutorial

app.py

@@ -3,9 +3,9 @@ from fastbook import load_learner
 from fast.ai.vision.all import *
 
 def predict(im):
-	im = PILImage.create(im)
-	pred, pred_idx, probs = model.predict(im)
-	return {labels[i]: float(probs[i]) for i in range (len(labels))}
+	im = im.reshape(1, 28*28)/255.0
+	pred = model.predict(im)[0]
+	return int(pred)
 
 labels = learn.dls.vocab
 
@@ -13,7 +13,7 @@ model = load_learner('./dr_model.pkl')
 
 demo = gr.Interface(
 	fn=predict,
-	inputs=gr.inputs.Image(shape=(28,28),invert_colors=True, source="canvas"),
-	outputs=gr.outputs.Label(num_top_classes=3),
-	interpretation="default")
+	inputs="sketchpad",
+	outputs = "label")
+
 demo.launch()