Upload app.py

app.py

@@ -9,6 +9,10 @@ learn = load_learner('model.pkl')
 
 categories = ('0','1','2','3','4','5','6','7','8','9')
 
+def get_y(r):
+    number = ''.join(filter(str.isdigit, str(r)))
+    return train_y[int(number)]
+
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
     return pred