Update app.py

app.py

@@ -58,10 +58,15 @@ def detect_objects(image, target_class, confidence_threshold, hf_token=None):
         results = object_detector.detect(image, hf_token)
         
         # Filter results for target class
+
         target_detections = []
         for detection in results:
-            if (detection['label'].lower() == target_class.lower() and 
-                detection['score'] > confidence_threshold):
+            label = detection.get('label')
+            if (
+                label is not None and
+                label.lower() == target_class.lower() and 
+                detection['score'] > confidence_threshold
+            ):
                 target_detections.append(detection)
         
         return target_detections