Hugging Face's logo

Spaces:
vsanchezn
/
minima
Sleeping

App Files Community
vsanchezn commited on
Commit
ec5b36c
·
1 Parent(s): 09e3cc0

fix categories

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. cat.jpg +0 -0
app.py CHANGED
@@ -17,7 +17,7 @@ def label_func(x):
17
  learn = load_learner('model.pkl')
18
 
19
  # Cell
20
- categories = ('Dog', 'Cat')
21
 
22
  def classify_image(img):
23
  pred, idx, probs = learn.predict(img)
 
17
  learn = load_learner('model.pkl')
18
 
19
  # Cell
20
+ categories = ('Cat', 'Dog')
21
 
22
  def classify_image(img):
23
  pred, idx, probs = learn.predict(img)
cat.jpg ADDED