Uploading Trashify box detection model app.py

README.md

@@ -4,10 +4,17 @@ emoji: 🗑️
 colorFrom: purple
 colorTo: blue
 sdk: gradio
-sdk_version: 4.40.0
 app_file: app.py
 pinned: false
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# 🚮 Trashify Object Detector
+
+Object detection demo to detect `trash`, `bin`, `hand`, `trash_arm`, `not_trash`, `not_bin`, `not_hand`. 
+
+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

app.py

@@ -107,7 +107,7 @@ def predict_on_image(image, conf_threshold):
         for item in target_items:
             if item not in class_name_text_labels:
                 missing_items.append(item)
-        return_string = f"Detected the following items: {class_name_text_labels}. But missing the following in order to get +1: {missing_items}. If this is an error, try altering the confidence threshold."
+        return_string = f"Detected the following items: {class_name_text_labels}. But missing the following in order to get +1: {missing_items}. If this is an error, try another image or altering the confidence threshold. Otherwise, the model may need to be updated with better data."
         
     # If all 3 trash, bin, hand occur = + 1
     if all_in_list(list_a=target_items, list_b=class_name_text_labels):