Update app.py

app.py

@@ -27,7 +27,7 @@ for name, model_file in MODEL_OPTIONS.items():
 
 # Custom MaskAnnotator for outline-only masks
 class OutlineMaskAnnotator:
-    def __init__(self, color: sv.Color = sv.Color.red(), thickness: int = 2):
+    def __init__(self, color: tuple = (255, 0, 0), thickness: int = 2):  # Changed from sv.Color.red()
         self.color = color
         self.thickness = thickness
         
@@ -46,7 +46,7 @@ class OutlineMaskAnnotator:
                 scene,
                 contours,
                 -1,
-                self.color.as_bgr(),
+                self.color,  # Directly use the color tuple
                 self.thickness
             )
         return scene
@@ -59,7 +59,7 @@ LABEL_ANNOTATOR = sv.LabelAnnotator(
     text_padding=2
 )
 MASK_ANNOTATOR = OutlineMaskAnnotator(
-    color=sv.Color.red(),
+    color=(255, 0, 0),  # Red color as BGR tuple
     thickness=2
 )
 
@@ -120,7 +120,7 @@ def detect_and_annotate(
     
     return annotated_image
 
-# Create Gradio interface (unchanged)
+# Rest of the Gradio interface remains exactly the same
 with gr.Blocks() as demo:
     gr.Markdown("# Medieval Manuscript Detection with YOLO")
     
@@ -177,7 +177,6 @@ with gr.Blocks() as demo:
     def clear():
         return None, None
 
-    # Connect buttons to functions
     detect_btn.click(
         process_image,
         inputs=[input_image, model_selector, conf_threshold, iou_threshold],