Update app.py

app.py

@@ -14,12 +14,11 @@ def is_cat(x): return x[0].isupper()
 
 learn = load_learner('animal-classifier.pkl')
 
-categories = ('cane','cavallo','elefante', 'farfalla', 'gallina', 'gatto', 'mucca', 'pecora', 'ragno', 'scoiattolo')
+# categories = ('cane','cavallo','elefante', 'farfalla', 'gallina', 'gatto', 'mucca', 'pecora', 'ragno', 'scoiattolo')
+categories = ('dog','horse','elephant', 'butterfly', 'chicken', 'cat', 'cow', 'sheep', 'spider', 'squirrel')
 
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
-    logger.info(pred)
-    logger.info(probs)
     return dict(zip(categories, map(float, probs)))
 
 image = gr.inputs.Image(shape=(192,192))