Demo

.gitignore

@@ -0,0 +1 @@
+.idea

Pipfile

@@ -0,0 +1,13 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[packages]
+gradio = "*"
+fastai = "*"
+
+[dev-packages]
+
+[requires]
+python_version = "3.10"

app.py

@@ -0,0 +1,32 @@
+
+import gradio as gr
+from fastai.vision.core import PILImage
+
+import fastai.vision.all as fav
+import os
+if os.name != 'posix':
+    print('Converting PosixPath to WindowsPath')
+    import pathlib
+    pathlib.PosixPath = pathlib.WindowsPath
+
+def is_cat(x): return x[0].isupper()  # Used by model
+learn = fav.load_learner('model.pkl')
+
+labels = learn.dls.vocab
+
+def predict(img):
+    img = PILImage.create(img)
+    pred,pred_idx,probs = learn.predict(img)
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+
+iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3))
+
+iface.launch()
+
+# def greet(name):
+#     return "Hello " + name + "!!"
+#
+# # iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+#
+# iface.launch()
+

model.pkl

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