| import gradio as gr | |
| from ultralytics import YOLO | |
| from PIL import Image | |
| model = YOLO("best.pt") | |
| def detect(image): | |
| results = model(image) | |
| return results[0].plot() | |
| demo = gr.Interface( | |
| fn=detect, | |
| inputs=gr.Camera(label="Take a photo with webcam"), # Enable camera input | |
| outputs=gr.Image(label="Detection Result"), | |
| title="Live Object Detection (Phone–Person–Bottle)", | |
| description="Click to capture from webcam and detect objects in real-time." | |
| ) | |
| demo.launch() | |