Upload folder using huggingface_hub

app.py

@@ -94,11 +94,17 @@ def add_label_to_image(image, prediction, confidence):
     rect_height = int(img_height * 0.15)
     rect_width = img_width - (padding * 2)
     
-    # Draw red rectangle
+    # Convert confidence percentage string to float
+    confidence_value = float(confidence.strip('%'))
+    
+    # Set color based on confidence (BGR format)
+    rect_color = (0, 255, 0) if confidence_value >= 80 else (0, 0, 255)  # Green if >=80%, Red otherwise
+    
+    # Draw colored rectangle
     cv2.rectangle(img_cv, 
                  (padding, padding), 
                  (padding + rect_width, padding + rect_height), 
-                 (0, 0, 255), 
+                 rect_color, 
                  -1)
     
     text = f"{prediction}: {confidence}"