Uploading Trashify V2 box detection model (with data augmentation) app.py

README.md

@@ -18,7 +18,16 @@ Used as example for encouraging people to cleanup their local area.
 
 If `trash`, `hand`, `bin` all detected = +1 point.
 
-* V1 = model trained *without* data augmentation
-* V2 = model trained *with* data augmentation
+## 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 - finish the README.md + update with links to materials

app.py

@@ -131,12 +131,14 @@ demo = gr.Interface(
     ],
     title="🚮 Trashify Object Detection Demo V2",
     description="""Help clean up your local area! Upload an image and get +1 if there is all of the following items detected: trash, bin, hand.
-    Model in V2 has been trained with data augmentation (tk - add link to model). 
+
+    The [model](https://huggingface.co/mrdbourke/detr_finetuned_trashify_box_detector_with_data_aug) in V2 has been trained with data augmentation preprocessing (color jitter, horizontal flipping) to improve robustness. 
     """,
     # 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
 )