Spaces:

7arunb
/

Cats

Runtime error

7arunb commited on Jul 28, 2022

Commit

a27dbe8

•

1 Parent(s): b3f3c73

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 inf = load_learner('LeopardCheetah (1).pkl')
-categories = ('Cheetah','Images','Leopard')
 def classify_img(img):
     pred, idx, probs = inf.predict(img)

 inf = load_learner('LeopardCheetah (1).pkl')
+categories = ('Cat', 'Cheetah', 'Other', 'Leopard', 'Lion', 'Snow Leopard', 'Tiger')
 def classify_img(img):
     pred, idx, probs = inf.predict(img)