import torch
import cv2
import numpy as np
import gradio as gr
from ultralytics import YOLO
from pillow import Image

# Modeli yükleme
model = YOLO("keremberke/yolov8n-pcb-defect-segmentation")

def detect_defects(image):
    results = model.predict(image, save=False)
    annotated_image = results[0].plot()  # Tahmin sonuçlarını çiz
    return annotated_image

iface = gr.Interface(
    fn=detect_defects,
    inputs=gr.Image(type="pil"),
    outputs=gr.Image(type="pil"),
    title="PCB Defect Segmentation"
)

iface.launch()