Deevyankar commited on
Commit
cbb436d
·
verified ·
1 Parent(s): 97dbadb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -0
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
+