Update app.py

app.py

@@ -7,14 +7,11 @@ from PIL import Image
 import time
 from fastai.vision.all import *
 from fastai.learner import load_learner
-
-def is_penguin(img):
-    return img.split('-')[0]
     
 #Load the Learner (Exported from ipnyb file with learn.export() )
-learn = load_learner('model.pkl')
+learn = load_learner('export.pkl')
 
-categories = ('Penguin', 'Puffin')
+categories = ('penguin', 'puffin', 'pufferfish')
 
 #Classify image
 def classify_image(cl_img):