from ultralytics import YOLO
import cv2 
import gradio as gr

file = "my_model.pt"
images = [["image1.jpg"],["image2.jpg"]]
model = YOLO(file)

def show_preds_image(image_path):
    image = cv2. imread(image_path)
    outputs = model.predict(source=image_path)
    results = outputs[0]. cpu().numpy()
    for i, det in enumerate(results.boxes.xyxy):
        cv2. rectangle(image,(int(det[0]), int(det[1])), (int(det[2]), int(det[3])),
                       thickness=6 , color=[0,0,255]
                       )
    return cv2. cvtColor (image, cv2.COLOR_BGR2RGB)


inputs_image = [
    gr.components.Image(type="filepath", label="Input Image"),
]
outputs_image = [
    gr.components.Image(type="numpy", label="Output Image"),
]

interface_image = gr.Interface(
    fn=show_preds_image,
    inputs=inputs_image,
    outputs=outputs_image,
    title="Plastic Detector",
    examples=images,
    cache_examples=False,
)

interface_image.launch()