new app

.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
+model.pkl filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,7 +1,19 @@
+from fastai.vision.all import *
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+def is_cat(x): return x[0].isUpper()
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+learn = load_learner('model.pkl')
+
+categories = ('Dog', 'Cat')
+
+def classify_image(img):
+    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 = ['dog.jpg', 'cat.jpg']
+
+intf = gr.Interface(fn=classify_image, inputs = image, outputs=label, examples=examples)
+intf.launch(inline=False, share=True)

flagged/log.csv

@@ -0,0 +1,5 @@
+'img','output','flag','username','timestamp'
+'img/0.jpg','{"Cat": 0.9831337928771973, "Dog": 0.016866201534867287}','','','2022-08-31 01:05:08.426331'
+'img/1.jpg','{"Cat": 0.9831337928771973, "Dog": 0.016866201534867287}','','','2022-08-31 01:05:10.402508'
+'img/2.jpg','{"Cat": 0.9831337928771973, "Dog": 0.016866201534867287}','','','2022-08-31 01:05:12.237047'
+'img/3.jpg','{"Cat": 0.9831337928771973, "Dog": 0.016866201534867287}','','','2022-08-31 01:05:14.566388'

model.pkl

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