deploy

.gitattributes

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

app.py

@@ -0,0 +1,29 @@
+from fastai.vision.all import *
+import gradio as gr
+import pathlib, os
+
+categories = ['daisy', 'dandelion', 'rose', 'sunflower', 'tulip']
+
+def classify_image(img):
+
+    learn = load_learner('flowers.pkl')
+    pred,idx,probs = learn.predict(img)
+    return dict(zip(categories, map(float, probs)))
+
+image = gr.inputs.Image(shape=(192,192))
+label = gr.outputs.Label()
+examples = [ 
+    'dandelion+seeds.jpg',
+    'common-dandelion-seeds-medical-herb-taraxacum-officinale.jpg',
+    'dandelion-seedhead.jpg',
+    'how-to-draw-sunflower.jpg',
+    'sunflower.jpg',
+    'sunflower1.jpg',
+    'tulip-drawing.jpg',
+    'broken-tulip-flower.jpg',
+    'rose01.jpg',
+    'rose02-blue.jpg'
+ ]
+
+iface = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+iface.launch()

flowers.pkl

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