zacpross commited on
Commit
390d0d9
1 Parent(s): 3706c52

Correct category order

Browse files
Files changed (4) hide show
  1. app.ipynb +0 -0
  2. app.py +1 -1
  3. cover_drive2.jpg +0 -0
  4. cut4.jpg +0 -0
app.ipynb CHANGED
The diff for this file is too large to render. See raw diff
app.py CHANGED
@@ -3,7 +3,7 @@ from fastai.vision.all import *
3
  import gradio as gr
4
  learn = load_learner('model.pkl')
5
  #|export
6
- categories = ('on side', 'off side')
7
 
8
  def classify_image(img):
9
  pred,idx,probs = learn.predict(img)
3
  import gradio as gr
4
  learn = load_learner('model.pkl')
5
  #|export
6
+ categories = ("off side", "on side")
7
 
8
  def classify_image(img):
9
  pred,idx,probs = learn.predict(img)
cover_drive2.jpg ADDED
cut4.jpg ADDED