Uploading Trashify box detection model v3 app.py with NMS post processing

README.md

@@ -18,8 +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
-* V3 = model trained *with* data augmentation & NMS ([Non Maximum Suppression](https://paperswithcode.com/method/non-maximum-suppression)) post processing step
+## 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

@@ -216,7 +216,8 @@ demo = gr.Interface(
     ],
     title="🚮 Trashify Object Detection Demo V3",
     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 V3 has been trained with data augmentation and has an additional post-processing step to filter classes for only the highest scoring box of each class. (tk - add link to model). 
+
+    The model in V3 is [same model](https://huggingface.co/mrdbourke/detr_finetuned_trashify_box_detector_with_data_aug) as in [V2](https://huggingface.co/spaces/mrdbourke/trashify_demo_v2) (trained with data augmentation) but has an additional post-processing step (NMS or [Non Maximum Suppression](https://paperswithcode.com/method/non-maximum-suppression)) to filter classes for only the highest scoring box of each class. 
     """,
     # Examples come in the form of a list of lists, where each inner list contains elements to prefill the `inputs` parameter with
     examples=[