Hugging Face's logo

Spaces:
EvanMarie
/
bears

App Files Community
EvanMarie commited on
Commit
eff928d
•
1 Parent(s): f3228d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
 
6
 
7
  learner_bears = load_learner('bears.pk1')
8
- categories = ['teddy bear', 'grizzly bear', 'black bear']
9
 
10
  def classify_bears(img):
11
  prediction, index, probability = learner_bears.predict(img)
 
5
 
6
 
7
  learner_bears = load_learner('bears.pk1')
8
+ categories = ['black bear', 'grizzly bear', 'teddy bear']
9
 
10
  def classify_bears(img):
11
  prediction, index, probability = learner_bears.predict(img)