File size: 664 Bytes
612a2c0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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()