| |
| """app.ipynb |
| |
| Automatically generated by Colab. |
| |
| Original file is located at |
| https://colab.research.google.com/drive/1nIR_ak2SXq9fwQmKpq4eupLwQU_SI4fn |
| """ |
|
|
| import gradio as gr |
| import torch |
| import cv2 |
| from ultralytics import YOLO |
| import numpy as np |
|
|
| model = YOLO("best.torchscript") |
|
|
| def predict(image): |
| results = model(image) |
| annotated = results[0].plot() |
| annotated_rgb = cv2.cvtColor(annotated, cv2.COLOR_BGR2RGB) |
| return annotated_rgb |
|
|
| iface = gr.Interface( |
| fn=predict, |
| inputs=gr.Image(type="numpy", label="Upload an image"), |
| outputs=gr.Image(type="numpy", label="Detected image"), |
| title="YOLOv8 Object Detection", |
| description="Upload an image to detect objects using a YOLOv8 model in TorchScript format.", |
| examples=None |
| ) |
|
|
| if __name__ == "__main__": |
| iface.launch() |
