add is_cat

app.py

@@ -5,6 +5,9 @@ learn = load_learner('export.pkl')
 
 labels = learn.dls.vocab
 
+def is_cat(x):
+    return 'cat' if (x.name)[0].isupper() else 'dog'
+
 def predict(img):
     img = PILImage.create(img)
     pred,pred_idx,probs = learn.predict(img)