Hugging Face's logo

Spaces:
7arunb
/
Cats
Runtime error

App Files Community
7arunb commited on
Commit
a27dbe8
·
1 Parent(s): b3f3c73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
7
 
8
  inf = load_learner('LeopardCheetah (1).pkl')
9
 
10
- categories = ('Cheetah','Images','Leopard')
11
 
12
  def classify_img(img):
13
  pred, idx, probs = inf.predict(img)
 
7
 
8
  inf = load_learner('LeopardCheetah (1).pkl')
9
 
10
+ categories = ('Cat', 'Cheetah', 'Other', 'Leopard', 'Lion', 'Snow Leopard', 'Tiger')
11
 
12
  def classify_img(img):
13
  pred, idx, probs = inf.predict(img)