🚀 Try it live on Hugging Face Spaces
This is a simple, interactive Streamlit app that lets you perform object detection using YOLOv5.
YOLO (You Only Look Once) is a real-time object detection system. It processes the image in one forward pass and predicts bounding boxes and class labels with remarkable speed.
In this app, we use the pre-trained yolov5s
model from Ultralytics, fine-tuned for speed and lightweight usage — perfect for CPU inference on Hugging Face Spaces.
import streamlit as st
from PIL import Image
import torch
from io import BytesIO
import numpy as np
import cv2
git clone https://github.com/96ibman/YOLOVision_Streamlit_HuggingFace.git
cd YOLOVision_Streamlit_HuggingFace
python -m venv venv
venv/Scripts/activate
pip install -r requirements.txt
streamlit run app.py