Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
| 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.
|
| 30 |
END_THRESHOLD = 0.1 # Probability to end an event
|
| 31 |
-
MOVING_AVG_WINDOW =
|
| 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 ---
|