vebie91 commited on
Commit
26d929a
1 Parent(s): 52489ed

update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,15 +5,15 @@ import pathlib
5
  plt = platform.system()
6
  if plt == 'Windows': pathlib.PosixPath = pathlib.WindowsPath
7
 
8
- def is_cat(x):
9
  return x[0].isupper()
10
 
11
  learn = load_learner('model.pkl')
12
 
13
  # labels = learn.dls.vocab
14
- #labels #['cat', 'dog']
15
 
16
- categories = ('cat', 'dog')
17
 
18
  def classify_image(img):
19
  pred, idx, probs = learn.predict(img)
 
5
  plt = platform.system()
6
  if plt == 'Windows': pathlib.PosixPath = pathlib.WindowsPath
7
 
8
+ def is_dog(x):
9
  return x[0].isupper()
10
 
11
  learn = load_learner('model.pkl')
12
 
13
  # labels = learn.dls.vocab
14
+ #labels #['dog', 'cat']
15
 
16
+ categories = ('dog', 'cat')
17
 
18
  def classify_image(img):
19
  pred, idx, probs = learn.predict(img)