Hugging Face's logo

Spaces:
black279
/
Littering-detector-v1
Sleeping

App Files Community
black279 commited on
Commit
787da7e
·
verified ·
1 Parent(s): 4f6ed2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,11 +7,11 @@ from ultralytics import YOLO
7
  from PIL import Image
8
 
9
  # ---------------- CONFIG ---------------- #
10
- CONF_THRESHOLD = 0.5
11
  COOLDOWN_TIME = 3 # seconds between alerts
12
  MODEL_PATH = "best.pt" # Place your model in the same directory
13
- FRAME_WIDTH = 640
14
- FRAME_HEIGHT = 480
15
  # ---------------------------------------- #
16
 
17
  # ---------------- MediaPipe Setup ---------------- #
 
7
  from PIL import Image
8
 
9
  # ---------------- CONFIG ---------------- #
10
+ CONF_THRESHOLD = 0.6
11
  COOLDOWN_TIME = 3 # seconds between alerts
12
  MODEL_PATH = "best.pt" # Place your model in the same directory
13
+ FRAME_WIDTH = 320
14
+ FRAME_HEIGHT = 240
15
  # ---------------------------------------- #
16
 
17
  # ---------------- MediaPipe Setup ---------------- #