changed predict function so it will not always show Ninjago, hopefully

app.py

@@ -4,14 +4,13 @@ import skimage
 
 learn = load_learner('export.pkl')
 
-categories = ('Lego Ninjago','Lego (non Ninjago)')
-
+labels = ('Lego Ninjago', 'Lego (non Ninjago)')
 
 
 def predict(img):
     img = PILImage.create(img)
     pred, pred_idx, probs = learn.predict(img)
-    return dict(zip(categories, map(float,probs)))
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 
 title = "Lego Classifier"