Spaces:

Raminnit
/

FADNet-Thermal-Detection

Sleeping

Raminnit commited on Apr 23

Commit

345e3ff

verified ·

1 Parent(s): d6e373c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ warnings.filterwarnings("ignore")
 # 0.  Constants & Paths  (edit these to match your environment)
 # ─────────────────────────────────────────────────────────────────────────────
 BASE_DIR   = pathlib.Path(__file__).parent
-CKPT_DIR   = BASE_DIR / "checkpoints"
 CHECKPOINTS = {
     "FADNet Finetune (Best)":  str(CKPT_DIR / "fadnet_finetune_best.pt"),

 # 0.  Constants & Paths  (edit these to match your environment)
 # ─────────────────────────────────────────────────────────────────────────────
 BASE_DIR   = pathlib.Path(__file__).parent
+CKPT_DIR   = BASE_DIR
 CHECKPOINTS = {
     "FADNet Finetune (Best)":  str(CKPT_DIR / "fadnet_finetune_best.pt"),