Spaces:
Sleeping
Sleeping
File size: 542 Bytes
8adebf2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | import gradio as gr
from ultralytics import YOLO
import cv2
import numpy as np
model = YOLO("best.pt")
def detect_objects(image):
results = model.predict(source=image, conf=0.25)
result_image = results[0].plot()
return result_image
demo = gr.Interface(
fn=detect_objects,
inputs=gr.Image(type="numpy"),
outputs=gr.Image(type="numpy"),
title="Hardware Scanner AI",
description="Upload une photo de ton PC Gamer. L'IA va détecter le GPU, la RAM, et le ventirad !",
)
demo.launch() |