Lord Walnuts commited on
Commit
bac6c08
1 Parent(s): 1b37606

changed categories

Browse files
Files changed (3) hide show
  1. app.py +1 -1
  2. mic1.jpg +0 -0
  3. tom1.jpg +0 -0
app.py CHANGED
@@ -3,7 +3,7 @@ from fastai.vision.all import *
3
 
4
  learn = load_learner('peaky.pkl')
5
 
6
- categories = ('Thomas Shelby','Arthur Shelby','John Shelby','Michael Gray','Ada Shelby','Polly Gray')
7
 
8
  def classify_image(img):
9
  pred,idx,probs = learn.predict(img)
 
3
 
4
  learn = load_learner('peaky.pkl')
5
 
6
+ categories = ('Ada Shelby','Arthur Shelby','John Shelby','Michael Gray','Polly Gray','Thomas Shelby')
7
 
8
  def classify_image(img):
9
  pred,idx,probs = learn.predict(img)
mic1.jpg ADDED
tom1.jpg ADDED