Update categories in app.py

app.py

@@ -3,13 +3,14 @@ from fastai.vision.all import *
 import os
 
 
-categories = ("Cristiano Ronaldo", "Lionel Messi")
+categories = ("Goat 🐐", "No Goat 😬")
 
 model = load_learner('model.pkl')
 
 def classify_image(img):
     pred, idx, probs = model.predict(img)
     print("pred: ", pred)
+    
     return dict(zip(categories, map(float, probs)))
 
 intf = gr.Interface(