Initial commit of 4way classifier.

.gitignore

@@ -1,2 +1,3 @@
 # vim buffer files
 *.swp 
+*.ipynb

.ipynb_checkpoints/app-checkpoint.py

@@ -0,0 +1,29 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: four-way-classifier.ipynb.
+
+# %% auto 0
+__all__ = ['learn', 'categories', 'image', 'label', 'examples', 'interf', 'classify_image']
+
+# %% four-way-classifier.ipynb 1
+from fastai.vision.all import *
+import gradio as gr
+import PIL
+import glob
+
+# %% four-way-classifier.ipynb 3
+learn = load_learner( 'simple-image-classifier.pkl' )
+
+# %% four-way-classifier.ipynb 5
+categories = [ 'bird', 'forest', 'otter', 'snake' ]
+
+def classify_image( img ):
+    pred, idx, probs = learn.predict( img )
+    # print( 'This is a %s'%pred )
+    return dict( zip( categories, map( float, probs ) ) )
+
+# %% four-way-classifier.ipynb 6
+image = gr.components.Image()
+label = gr.components.Label()
+examples = glob.glob( './resized-*.jpg' )
+
+interf = gr.Interface( fn=classify_image, inputs=image, outputs=label, examples=examples, allow_flagging='manual' )
+interf.launch( inline=True )

app.py

@@ -1,7 +1,44 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: four-way-classifier.ipynb.
+
+# %% auto 0
+__all__ = ['learn', 'categories', 'description', 'image', 'label', 'examples', 'interf', 'classify_image']
+
+# %% four-way-classifier.ipynb 1
+from fastai.vision.all import *
 import gradio as gr
+import PIL
+import glob
+
+# %% four-way-classifier.ipynb 3
+learn = load_learner( 'simple-image-classifier.pkl' )
+
+# %% four-way-classifier.ipynb 5
+categories = [ 'bird', 'forest', 'otter', 'snake' ]
+
+def classify_image( img ):
+    pred, idx, probs = learn.predict( img )
+    # print( 'This is a %s'%pred )
+    return dict( zip( categories, map( float, probs ) ) )
+
+# %% four-way-classifier.ipynb 7
+description='''
+A simple 4-way classifier that categorizes images as 'snake', 'bird', 'otter' or 'forest'. 
+Refined from an initial ResNet18 model downloaded from HuggingFace.
 
-def greet( name ):
-    return 'Hello %s!!'%name
+**DISCLAIMER**: the images here are merely for demonstration purposes. I don't own any of them 
+and I'm not making money from them.  
+'''
+image = gr.components.Image()
+label = gr.components.Label()
+examples = glob.glob( './*.jpg' )
 
-iface = gr.Interface( fn=greet, inputs='text', outputs='text' )
-iface.launch()
+interf = gr.Interface( 
+    title='Simple 4-way image classifier', 
+    description=description, 
+    fn=classify_image, 
+    inputs=image, 
+    outputs=label, 
+    examples=examples, 
+    allow_flagging='manual' 
+)
+interf.launch( inline=True )

simple-image-classifier.pkl

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