File size: 939 Bytes
822c6af
 
 
 
 
 
 
 
 
 
cbb436d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
---
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