vaishnavkdinesh/detect_drones

1import cv22import gradio as gr3from ultralytics import YOLO4 5model = YOLO("best.pt")6 7def predict_drone(img):8  img_copy = img.copy()9  result = model(img_copy)10  for data in result[0].boxes.data:11    xmin,ymin,xmax,ymax = int(data[0]),int(data[1]),int(data[2]),int(data[3])12    cv2.rectangle(img_copy,(xmin,ymin),(xmax,ymax),(0,255,0),2)13  return img_copy14 15 16image_interface=gr.Interface(fn = predict_drone,inputs="image",outputs="image")17image_interface.launch()