jrnold commited on
Commit
48efa13
Β·
1 Parent(s): cbf3da8
examples/Bengal_102.jpg β†’ Bengal_102.jpg RENAMED
File without changes
examples/Sphynx_143.jpg β†’ Sphynx_143.jpg RENAMED
File without changes
app.py CHANGED
@@ -11,7 +11,7 @@ def is_cat(x) -> bool:
11
 
12
  learn = load_learner("model.pkl")
13
 
14
- categories = learn.dls.vocab
15
 
16
  def classify_image(img):
17
  _, _, probs = learn.predict(img)
 
11
 
12
  learn = load_learner("model.pkl")
13
 
14
+ categories = ("Dog", "Cat")
15
 
16
  def classify_image(img):
17
  _, _, probs = learn.predict(img)
examples/chihuahua_43.jpg β†’ chihuahua_43.jpg RENAMED
File without changes
examples/english_setter_15.jpg β†’ english_setter_15.jpg RENAMED
File without changes
examples/havanese_129.jpg β†’ havanese_129.jpg RENAMED
File without changes
examples/japanese_chin_83.jpg β†’ japanese_chin_83.jpg RENAMED
File without changes