anchala23/drone_flagger

1import cv22from ultralytics import YOLO3import gradio as gr4model = YOLO("best.pt")5def predict_drone(img):6  results = model.predict(img)7  for data in results[0].boxes.data:8    xmin, ymin, xmax, ymax = int(data[0]), int(data[1]), int(data[2]), int(data[3])9    cv2.rectangle(img, (xmin, ymin), (xmax, ymax), (0, 255, 0), 2)10  return img11interface = gr.Interface(fn=predict_drone, inputs="image", outputs="image")12interface.launch()