Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+aura_farmer_1.jpg filter=lfs diff=lfs merge=lfs -text
+aura_farmer_2.jpg filter=lfs diff=lfs merge=lfs -text
+aura_farmer_3.jpg filter=lfs diff=lfs merge=lfs -text
+aura_farmer_4.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,49 @@
+import gradio as gr
+from ultralytics import YOLO
+import torch
+
+model_id = "mosesb/best-comic-panel-detection"
+model = YOLO("best.pt")
+
+def detect_panels(pil_image):
+    """
+    Takes a PIL image, runs YOLOv12 object detection to find comic panels,
+    and returns the annotated image with bounding boxes.
+    """
+    results = model.predict(pil_image, verbose=False)
+
+    # The .plot() method returns a BGR NumPy array with the detections plotted on the original image.
+    annotated_image = results[0].plot()
+
+    # Gradio's gr.Image component expects an RGB image. The .plot() method
+    # returns a BGR image (common in OpenCV), so we convert it.
+    annotated_image_rgb = annotated_image[..., ::-1]
+
+    return annotated_image_rgb
+
+
+# --- Gradio Interface ---
+title = "YOLOv12 Comic Panel Detection"
+description = """
+This demo showcases a **YOLOv12 object detection model** that has been fine-tuned to detect panels in comic book pages.
+Upload an image of a comic page, and the model will draw bounding boxes around each detected panel. 
+This can be a useful first step for downstream tasks like Optical Character Recognition (OCR) or character analysis within comics.
+"""
+
+article = f"""
+<div style='text-align: center;'>
+    <p style='text-align: center'>Model loaded from <a href='https://huggingface.co/{model_id}' target='_blank'>{model_id}</a></p>
+    <p style='text-align: center'>For more details on the training process, check out the project repository: <a href='https://github.com/mosesab/YOLOV12-Comic-Panel-Detection/blob/main/comic-boundary-detection.ipynb' target='_blank'>Comic Boundary Detection</a></p>
+</div>
+"""
+
+gr.Interface(
+    fn=detect_panels,
+    inputs=gr.Image(type="pil", label="Upload Comic Page Image"),
+    outputs=gr.Image(type="pil", label="Detected Panels"),
+    title=title,
+    description=description,
+    article=article,
+    examples=[ "aura_farmer_1.jpg", "aura_farmer_2.jpg", "aura_farmer_3.jpg", "aura_farmer_4.jpg" ],
+    allow_flagging="auto"
+).launch()

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:566e9ff59ec146afb695eb48a8aaae983b43bb72c3a57f63afafe12f3b349af4
+size 119112954

last.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ee64422e2d2cea6b37dbd211c7e24137c429030c2e11b85bcd8dbeaa2f7e9295
+size 119112954

requirements.txt

@@ -0,0 +1,3 @@
+torch #2.7.1
+ultralytics #8.3.160
+gradio #5.35.0