import gradio as gr
import torch
from ultralytics import YOLO

# Load YOLO model
model = YOLO("yolov9c-seg.pt")

# Define function to perform prediction
def predict_image(image):
    result = model.predict(image, show=True, save=True)
    return result

# Create Gradio interface
image_input = gr.inputs.Image()
gr.Interface(predict_image, image_input, "image").launch()