Hugging Face's logo

Spaces:
anglesplato
/
minimal
Sleeping

App Files Community
anglesplato commited on
Commit
34cdeb7
·
verified ·
1 Parent(s): 08caf8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  from fastai.vision.all import *
4
 
5
  categories = ('Dog', 'Cat')
6
- learn = load_learner('model.pk1')
7
  def classify_image(img):
8
  pred , idx , probs = learn.predict(img)
9
  return dict(zip(categories , map(float,probs)))
 
3
  from fastai.vision.all import *
4
 
5
  categories = ('Dog', 'Cat')
6
+ learn = load_learner('model.pkl')
7
  def classify_image(img):
8
  pred , idx , probs = learn.predict(img)
9
  return dict(zip(categories , map(float,probs)))