Uploading Trashify box detection model app.py

README.md

@@ -10,7 +10,7 @@ pinned: false
 license: apache-2.0
 ---
 
-# 🚮 Trashify Object Detector
+# 🚮 Trashify Object Detector V1 
 
 Object detection demo to detect `trash`, `bin`, `hand`, `trash_arm`, `not_trash`, `not_bin`, `not_hand`. 
 
@@ -18,4 +18,16 @@ Used as example for encouraging people to cleanup their local area.
 
 If `trash`, `hand`, `bin` all detected = +1 point.
 
-TK - finish the README.md + update with links to materials
+## Dataset
+
+All Trashify models are trained on a custom hand-labelled dataset of people picking up trash and placing it in a bin.
+
+The dataset can be found on Hugging Face as [`mrdbourke/trashify_manual_labelled_images`](https://huggingface.co/datasets/mrdbourke/trashify_manual_labelled_images).
+
+## Demos
+
+* [V1](https://huggingface.co/spaces/mrdbourke/trashify_demo_v1) = Fine-tuned DETR model trained *without* data augmentation.
+* [V2](https://huggingface.co/spaces/mrdbourke/trashify_demo_v2) = Fine-tuned DETR model trained *with* data augmentation.
+* [V3](https://huggingface.co/spaces/mrdbourke/trashify_demo_v3) = Fine-tuned DETR model trained *with* data augmentation (same as V2) with an NMS (Non Maximum Suppression) post-processing step.
+
+TK - add links to resources to learn more

app.py

@@ -133,7 +133,8 @@ demo = gr.Interface(
     # Examples come in the form of a list of lists, where each inner list contains elements to prefill the `inputs` parameter with
     examples=[
         ["examples/trashify_example_1.jpeg", 0.25],
-        ["examples/trashify_example_2.jpeg", 0.25]
+        ["examples/trashify_example_2.jpeg", 0.25],
+        ["examples/trashify_example_3.jpeg", 0.25],
     ],
     cache_examples=True
 )