Spaces:
Sleeping
Sleeping
Update README.md
Browse files
README.md
CHANGED
|
@@ -8,3 +8,35 @@ pinned: false
|
|
| 8 |
---
|
| 9 |
|
| 10 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
| 11 |
+
|
| 12 |
+
# 🎥 Multi-Face Attention Detector (YOLOv7, Streamlit, GPU)
|
| 13 |
+
|
| 14 |
+
This app detects attention in multiple faces using YOLOv7 + mediapipe + a custom ML model.
|
| 15 |
+
|
| 16 |
+
## 🚀 Setup on Hugging Face
|
| 17 |
+
|
| 18 |
+
1. Create a Space → choose Blank Docker template.
|
| 19 |
+
2. Upload:
|
| 20 |
+
- app.py
|
| 21 |
+
- requirements.txt
|
| 22 |
+
- Dockerfile
|
| 23 |
+
- model.pkl (your ML model)
|
| 24 |
+
- example_input.jpg (optional test image)
|
| 25 |
+
|
| 26 |
+
3. Enable GPU hardware.
|
| 27 |
+
|
| 28 |
+
4. Deploy!
|
| 29 |
+
|
| 30 |
+
## ✅ Features
|
| 31 |
+
|
| 32 |
+
- YOLOv7 GPU face + phone detection
|
| 33 |
+
- Mediapipe fallback (via mediapipe-nightly)
|
| 34 |
+
- OpenCV video processing
|
| 35 |
+
- Streamlit dashboard with charts, logs, CSV download
|
| 36 |
+
|
| 37 |
+
## 💡 Notes
|
| 38 |
+
|
| 39 |
+
- Base image: `nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04`
|
| 40 |
+
- Compatible with `torch==2.1.2+cu118`
|
| 41 |
+
- Includes placeholder model.pkl for demo
|
| 42 |
+
|