initial commit

README.md

@@ -1,13 +1,14 @@
 ---
 title: Poodle Or Doodle
-emoji: 📚
+emoji: 🐶
 colorFrom: red
-colorTo: indigo
+colorTo: yellow
 sdk: gradio
 sdk_version: 4.40.0
 app_file: app.py
 pinned: false
 license: mit
+short_description: Upload an image and see if it's a poodle or doodle.
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+A small Gradio app backed by a fine-tuned resnet18 model for predicting if you're looking at a standard poodle or a labradoodle.

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+from PIL import Image
+from fastai.vision.all import *
+
+learn = load_learner("model.pkl")
+
+
+def predict(img):
+    # use PIL to resize the image to 224x224
+    img = Image.fromarray(img)
+    pred, pred_idx, probs = learn.predict(img)
+    # if the prediction contains poodle, return poodle
+    if "poodle" in pred:
+        return "poodle"
+    return "doodle"
+
+
+title = "Poodle or Doodle"
+description = "A resnet18 model fine tuned on 400 images of poodles and labradoodles. You can predict using the default image or upload an image an image of your own to see if you're looking at a poodle or doodle."
+
+gr.Interface(
+    fn=predict,
+    inputs=gr.Image(value="labradoodle.jpg"),
+    outputs=gr.Label(),
+    title=title,
+    description=description,
+    allow_flagging="never",
+).launch()

model.pkl

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

pyproject.toml

@@ -0,0 +1,19 @@
+[tool.poetry]
+name = "poodle-or-doodle"
+version = "0.1.0"
+description = "A small Gradio app for seeing if the image is a poodle or labradoodle backed by a fine-tuned resnet18 model"
+authors = ["jsulz <[email protected]>"]
+license = "MIT"
+readme = "README.md"
+package-mode = false
+
+[tool.poetry.dependencies]
+python = "3.12.2"
+gradio = "4.40.0"
+fastai = "^2.7.16"
+pillow = "^10.4.0"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"