Add detection_items.py to list YOLOS detectable object names and update README.md

Author: muhammad-fiaz

README.md

@@ -71,7 +71,7 @@ phone,42,38,20
 bottle,100,0,0
 ```
 
-- `label`: The item name (must match YOLOS labels for detection).
+- `label`: The item name (must match YOLOS labels; you can list them by running `python detection_items.py`)
 - `plastic`, `metal`, `glass`: Composition percentages (should sum to ~100).
 
 ---

detection_items.py

@@ -0,0 +1,14 @@
+import os
+
+os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
+from transformers import YolosForObjectDetection
+
+# Load the model config
+model = YolosForObjectDetection.from_pretrained("hustvl/yolos-base")
+
+# Get all detectable object names
+object_names = list(model.config.id2label.values())
+
+# Display the object names
+for name in object_names:
+    print(name)