JAI / README.md
Deevyankar's picture
Update README.md
cbb436d verified
---
title: JAI
emoji: πŸš€
colorFrom: indigo
colorTo: gray
sdk: docker
pinned: false
---
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
# πŸŽ₯ Multi-Face Attention Detector (YOLOv7, Streamlit, GPU)
This app detects attention in multiple faces using YOLOv7 + mediapipe + a custom ML model.
## πŸš€ Setup on Hugging Face
1. Create a Space β†’ choose Blank Docker template.
2. Upload:
- app.py
- requirements.txt
- Dockerfile
- model.pkl (your ML model)
- example_input.jpg (optional test image)
3. Enable GPU hardware.
4. Deploy!
## βœ… Features
- YOLOv7 GPU face + phone detection
- Mediapipe fallback (via mediapipe-nightly)
- OpenCV video processing
- Streamlit dashboard with charts, logs, CSV download
## πŸ’‘ Notes
- Base image: `nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04`
- Compatible with `torch==2.1.2+cu118`
- Includes placeholder model.pkl for demo