Create app.py

app.py

@@ -0,0 +1,35 @@
+import gradio as gr
+import torch
+from ultralytics import YOLO
+import cv2
+import numpy as np
+from PIL import Image
+
+# Load trained YOLO11 model
+model_path = "model.pt"  # Ensure this is uploaded to your HF Space
+model = YOLO(model_path)
+
+# Define inference function
+def segment_card(image):
+    image = np.array(image)  # Convert PIL image to numpy array
+    results = model(image)   # Run YOLO model
+
+    # Extract bounding boxes
+    for result in results:
+        for box in result.boxes:
+            x1, y1, x2, y2 = map(int, box.xyxy[0])  # Bounding box coords
+            cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)  # Draw box
+
+    return Image.fromarray(image)  # Convert back to PIL image
+
+# Create Gradio UI
+iface = gr.Interface(
+    fn=segment_card,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="MTG Card Segmentation with YOLO11",
+    description="Upload a Magic: The Gathering card image, and the model will segment key visual elements."
+)
+
+# Launch the app
+iface.launch()