Upload app.py

app.py

@@ -3,14 +3,16 @@ from fastai.vision.all import *
 
 __all__ = ['NumberClass', 'learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
 
+def get_y(r):
+    number = ''.join(filter(str.isdigit, str(r)))
+    return train_y[int(number)]
+
 def NumberClass(x): return x[0].isupper()
 
 learn = load_learner('model.pkl')
 
 categories = ('0','1','2','3','4','5','6','7','8','9')
 
-def get_y(r): pass
-
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
     return pred