import torch
import cv2
import numpy as np
import gradio as gr
from ultralytics import YOLO
from PIL import Image
from huggingface_hub import hf_hub_download

# Hugging Face'ten YOLO modelini indir
model_path = hf_hub_download("keremberke/yolov8n-pcb-defect-segmentation", "best.pt")

# Modeli yükle
model = YOLO(model_path)

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()