foreversheikh commited on
Commit
109d7a6
·
verified ·
1 Parent(s): eb3db51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = "pretrained/c3d.pickle"
38
- AD_MODEL_PATH = "exps/c3d/models/epoch_80000.pt"
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