Spaces:
No application file
No application file
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,9 +34,9 @@ socketio = SocketIO(app, async_mode='eventlet')
|
|
| 34 |
# ---- Global Config & Model Loading ----
|
| 35 |
print("[INFO] Loading models...")
|
| 36 |
DEVICE = get_torch_device()
|
| 37 |
-
FEATURE_EXTRACTOR_PATH = "
|
| 38 |
-
AD_MODEL_PATH = "
|
| 39 |
-
YOLO_MODEL_PATH = "yolo_my_model.pt"
|
| 40 |
SAVE_DIR = "outputs/anomaly_frames"
|
| 41 |
ANOMALY_THRESHOLD = 0.4
|
| 42 |
COOLDOWN_SECS = 60.0 # <--- This enforces the 60 second anomaly cooldown
|
|
|
|
| 34 |
# ---- Global Config & Model Loading ----
|
| 35 |
print("[INFO] Loading models...")
|
| 36 |
DEVICE = get_torch_device()
|
| 37 |
+
FEATURE_EXTRACTOR_PATH = "models/c3d.pickle"
|
| 38 |
+
AD_MODEL_PATH = "models/epoch_80000.pt"
|
| 39 |
+
YOLO_MODEL_PATH = "models/yolo_my_model.pt"
|
| 40 |
SAVE_DIR = "outputs/anomaly_frames"
|
| 41 |
ANOMALY_THRESHOLD = 0.4
|
| 42 |
COOLDOWN_SECS = 60.0 # <--- This enforces the 60 second anomaly cooldown
|