init

export.pkl

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

main.py

@@ -0,0 +1,32 @@
+import gradio as gr
+from PIL import Image
+from fastai.vision.all import *
+
+def predict_image(image):
+    learner = learner_load('export.pkl')
+    img = PILImage.create(image)
+    pred = learner.predict(img)
+    return pred[0]
+
+
+def create_interface():
+    
+    image_input = gr.inputs.Image()
+
+    output = gr.outputs.Text()
+
+    iface = gr.Interface(
+        fn=predict_image,
+        inputs=image_input,
+        outputs=output,
+        capture_session=True,
+        title="Cat or Dog Classifier",
+        description="Upload an image to classify whether it's a cat or a dog."
+    )
+    return iface
+
+if __name__ == "__main__":
+    iface = create_interface()
+    iface.launch()
+
+

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+pillow
+fastai