initial commit

.gitattributes

@@ -32,3 +32,12 @@ 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
+clean-02.JPG filter=lfs diff=lfs merge=lfs -text
+clean-03.JPG filter=lfs diff=lfs merge=lfs -text
+dirty-02.JPG filter=lfs diff=lfs merge=lfs -text
+dirty-03.JPG filter=lfs diff=lfs merge=lfs -text
+seiba-edisa-apple-bins.pkl filter=lfs diff=lfs merge=lfs -text
+clean-01.JPG filter=lfs diff=lfs merge=lfs -text
+clean-04.JPG filter=lfs diff=lfs merge=lfs -text
+dirty-01.JPG filter=lfs diff=lfs merge=lfs -text
+dirty-04.JPG filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+from fastai.vision.all import *
+
+learn = load_learner('seiba-edisa-apple-bins.pkl')
+
+categories = ['Clean', 'Dirty']
+def predict(img):
+    # img = PILImage.create(img)
+    _,_,probs = learn.predict(img)
+    return dict(zip(categories, map(float,probs)))
+
+title = "Apple bins classifier"
+examples = ['clean-01.JPG','clean-02.JPG','clean-03.JPG','clean-04.JPG','dirty-01.JPG','dirty-02.JPG','dirty-03.JPG','dirty-04.JPG']
+interpretation='default'
+enable_queue=True
+
+gr.Interface(fn=predict,
+             inputs=gr.inputs.Image(shape=(192, 192)),
+             outputs=gr.outputs.Label(num_top_classes=2),
+             title=title,
+             examples=examples).launch(inline=False)

requirements.txt

@@ -0,0 +1,2 @@
+fastai
+gradio

seiba-edisa-apple-bins.pkl

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