Hugging Face's logo

Spaces:
JediHustle
/
beartector
Sleeping

App Files Community
JediHustle commited on
Commit
e45e026
·
1 Parent(s): c76b17e

Added Panda bear

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. beartector.pkl +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
  learn = load_learner('beartector.pkl')
7
 
8
  #|export
9
- categories = ('Black Bear', 'Grizzly Bear', 'Koala Bear', 'Polar Bear', 'Teddy Bear')
10
  def classify_image(img):
11
  pred,idx,probs = learn.predict(img)
12
  return dict(zip(categories, map(float,probs)))
 
6
  learn = load_learner('beartector.pkl')
7
 
8
  #|export
9
+ categories = ('Black Bear', 'Grizzly Bear', 'Koala Bear', 'Panda Bear', 'Polar Bear', 'Teddy Bear')
10
  def classify_image(img):
11
  pred,idx,probs = learn.predict(img)
12
  return dict(zip(categories, map(float,probs)))
beartector.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4566fc5d4b4402e8aaa5cb3c3cb3e70145eb49a440976e661317d3556d22f478
3
- size 87490799
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef8ee4b8aee49d3493bcd6ec4cfac8a523495aaa602773846d61e01a3394a820
3
+ size 87493615