Upload 2 files

classifier.py

@@ -0,0 +1,17 @@
+from fastai.vision.all import *
+import gradio as gr
+
+class Classifier:
+    def __init__(self, model_path):
+        self.learn = load_learner(model_path)
+
+    def predict(self, image):
+        pred, pred_idx, probs = self.learn.predict(image)
+        return {self.learn.dls.vocab[i]: float(p) for i, p in enumerate(probs)}
+    
+image = gr.inputs.Image(shape=(128, 128))
+label = gr.outputs.Label()
+classifer = Classifier(model_path="model.pkl")
+
+interface = gr.Interface(fn=classifer.predict, inputs=image, outputs=label)
+interface.launch(inline=False)

model.pkl

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