let's deploy to huggingface spaces5

app.py

@@ -18,13 +18,17 @@ def showcase(imgs):
         download_url(url[0], dest, show_progress=False)
 
 
+def label_func(f):
+    return f.split("_")[0]
+
+
 learner = load_learner("model3.pk3")
 
 labels = learner.dls.vocab
 
 
-def predict(img):
-    img = PILImage.create(img)
+def predict(image):
+    img = PILImage.create(image)
     pred, pred_idx, probs = learner.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}