Verocayden commited on
Commit
3dcc988
·
verified ·
1 Parent(s): 8b4af57

Added is_cat function

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -5,6 +5,10 @@ from fastai.vision.all import *
5
 
6
  MODEL_FILENAME = 'model.pkl'
7
 
 
 
 
 
8
  learn = load_learner(MODEL_FILENAME)
9
  categories = ('Dog', 'Cat')
10
 
 
5
 
6
  MODEL_FILENAME = 'model.pkl'
7
 
8
+
9
+
10
+ def is_cat(x): return x[0].isupper()
11
+
12
  learn = load_learner(MODEL_FILENAME)
13
  categories = ('Dog', 'Cat')
14