update to multilabel model

.gitattributes

@@ -33,4 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text

app.py

@@ -4,16 +4,15 @@ import os
 os.environ["OMP_NUM_THREADS"] = "1"
 os.environ["MKL_NUM_THREADS"] = "1"
 
-
-learn = load_learner('model.pkl', cpu=True)
+learn = load_learner('bears.pkl', cpu=True)
 learn.model.eval()
 
-categories = ('black', 'grizzly', 'teddy', 'white')
-
 def classify_image(img):
     img = PILImage.create(img)
     pred, idx, probs = learn.predict(img)
-    return dict(zip(categories, map(float, probs)))
+    results = {label: float(p) for label, p in zip(learn.dls.vocab, probs)}
+    results['no bear'] = max(0, 1.0 - max(results.values()))
+    return results
 
 image = gr.Image(width=192, height=192, type="pil")
 label = gr.Label()

bears.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:20d180091fddb59a97e84c3c6e6d34ce2cc9c8f5333439a729a0f82efa7672e9
+size 102898651