mmzoellner commited on
Commit
4b52ccc
·
verified ·
1 Parent(s): b8bc747

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,7 +15,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
  # --- Config ---
16
  SAMPLE_RATE = 16000
17
  CHUNK_SEC = 2
18
- STRIDE_SEC = 0.2 # 🔥 stride = 0.2s → ~5 preds per second
19
  MAX_SEC = 60
20
  IMG_SIZE = 224
21
  N_MELS = 128
@@ -26,9 +26,9 @@ HF_REPO_ID = "mmzoellner/MobileNet_ATD"
26
  MOBILENET_WEIGHTS_FILENAME = "mobilenet_v3_small_fold2.pt"
27
 
28
  # --- Event Detection Thresholds & Smoothing ---
29
- START_THRESHOLD = 0.5 # Probability to start an event
30
  END_THRESHOLD = 0.1 # Probability to end an event
31
- MOVING_AVG_WINDOW = 5 # Window size for smoothing (5 samples = 1 sec of preds)
32
 
33
 
34
  # --- Audio -> Mel-dB -> 3ch 224x224 ---
 
15
  # --- Config ---
16
  SAMPLE_RATE = 16000
17
  CHUNK_SEC = 2
18
+ STRIDE_SEC = 0.1 # 🔥 stride = 0.2s → ~5 preds per second
19
  MAX_SEC = 60
20
  IMG_SIZE = 224
21
  N_MELS = 128
 
26
  MOBILENET_WEIGHTS_FILENAME = "mobilenet_v3_small_fold2.pt"
27
 
28
  # --- Event Detection Thresholds & Smoothing ---
29
+ START_THRESHOLD = 0.3 # Probability to start an event
30
  END_THRESHOLD = 0.1 # Probability to end an event
31
+ MOVING_AVG_WINDOW = 10 # Window size for smoothing (5 samples = 1 sec of preds)
32
 
33
 
34
  # --- Audio -> Mel-dB -> 3ch 224x224 ---