init

.gitignore

@@ -0,0 +1 @@
+venv

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app/export.pkl

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

app/main.py

@@ -0,0 +1,34 @@
+import gradio as gr
+import tensorflow as tf
+from PIL import Image
+import numpy as np
+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()
+
+

artifacts/reptile_weight_79.pth

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

requirements.txt

@@ -0,0 +1,4 @@
+uvicorn
+gradio
+nicegui
+fastai