Spaces:
Sleeping
Sleeping
File size: 527 Bytes
369be04 a26946d b35c28c 417937d f544e86 417937d f6600f7 417937d f6600f7 417937d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from ultralytics import YOLO
from PIL import Image
import gradio as gr
import numpy as np
def function(image):
model = YOLO('best_yolo_2.pt')
image = np.array(image)
results = model.predict(image)
img=Image.fromarray(results[0].plot())
num_result=len(results[0].boxes.cls)
names_output=[]
for i in range(num_result):
name=results[0].names[int(results[0].boxes.cls[i])]
names_output.append(name)
return names_output,img
demo=gr.Interface(fn=function,inputs="image",outputs=["text","image"])
demo.launch() |