Added code to run model on hugging face

README.md

@@ -1,8 +1,8 @@
 ---
-title: Doodoodetective
-emoji: 🐨
+title: Doo-doo Detective
+emoji: 📊
 colorFrom: indigo
-colorTo: green
+colorTo: red
 sdk: gradio
 sdk_version: 3.3.1
 app_file: app.py

app.py

@@ -0,0 +1,21 @@
+from fastai.vision.all import *
+
+learn = load_learner('poop.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))}
+
+import gradio as gr
+
+
+# add a title and description
+title = "Is this a photo of a doing his business? Or just a dog hanging out?"
+desc = "My dog Kona has a nasty habit of doing her business in a very particular spot in the upstairs of our house. Using this model, I'll be able to aim a camera at that spot and send a text alert whenever she does this, so I'll have a better chance of correcting the bad behavior in the moment, which is essential to training a dog."
+
+iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3), title=title, description=desc)
+
+iface.launch()
+

requirements.txt

@@ -0,0 +1 @@
+fastai