Tracking examples as binary files via git lfs

.gitattributes

@@ -32,3 +32,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
+examples/*.* filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,29 +1,12 @@
-# import gradio as gr
-#
-# def greet(name):
-#     return "Hello " + name + "!!"
-#
-# iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-# iface.launch()
-
-# Cell
 from fastai.vision.all import *
 import gradio as gr
 import timm
 import dill
+import os
 
-
-def is_catan(x):
-    # print(x[:5])
-    # print(x[:5] == 'Catan')
-    return x[:5] == 'Catan'
-
-
-# Cell
-learn = load_learner('./models/catan-model-paperspace-3.pkl', pickle_module=dill)
+learn = load_learner('./models/catan-model-paperspace-5.pkl', pickle_module=dill)
 # learn = load_learner('catan-model.pkl', pickle_module=dill)
 
-# Cell
 # categories = learn.dls.vocab
 categories = ('Not Catan', 'Catan')
 
@@ -36,7 +19,8 @@ def classify_image(img):
 # Cell
 image = gr.inputs.Image(shape=(192, 192))
 label = gr.outputs.Label()
-examples = ['board-game-catan-IMG_4671.jpg', 'photo-of-macbook-catan-IMG_4817.jpg']
+examples_dir_path = './examples/'
+examples = [(examples_dir_path + filename) for filename in os.listdir(examples_dir_path) if filename[:1] != '.']
 
 # Cell
 intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)

models/catan-model-paperspace-5.pkl

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