FractureVision / app.py
DHEIVER's picture
Create app.py
c38745d verified
raw
history blame
782 Bytes
# 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()