cvergnes commited on
Commit
69b3097
1 Parent(s): 4086e03
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ label = gr.outputs.Label()
8
  examples = ['bmw_750.jpeg']
9
  categories = ['bmw car', 'mercedes car']
10
 
11
- learn = load_learner("export.kpl")
12
 
13
  def classify_image(img):
14
  pred, _, prob = learn.predict(img)
 
8
  examples = ['bmw_750.jpeg']
9
  categories = ['bmw car', 'mercedes car']
10
 
11
+ learn = load_learner("export.pkl")
12
 
13
  def classify_image(img):
14
  pred, _, prob = learn.predict(img)