# from fastapi import FastAPI, File, UploadFile
# from fastapi.responses import StreamingResponse
from io import BytesIO
import cv2
import numpy as np
import gradio as gr
from ultralytics import YOLO

# Inicializar o modelo YOLO
model = YOLO("/content/best.pt")

# Definir a função para detecção de objetos
def detect_objects(image):
    # Realizar a detecção de objetos com YOLOv8
    detections = model.predict(image)
    # Renderizar as detecções na imagem
    rendered_image = detections.render()
    return rendered_image

# Criar a interface Gradio
app = gr.Interface(
    fn=detect_objects,
    inputs="image",
    outputs="image",
    title="YOLOv8 Object Detection",
    description="Detect objects in images using YOLOv8.",
)

# Executar o aplicativo
app.launch()