Hugging Face's logo

Spaces:
telecomadm1145
/
AIDetectV2
Running

App Files Community
telecomadm1145 commited on
Commit
c101d2c
·
verified ·
1 Parent(s): 77d9851

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ from pytorch_grad_cam.utils.image import show_cam_on_image
19
 
20
  # --- Configuration ---------------------------------------------------------
21
  REPO_ID = "telecomadm1145/swin-ai-detection"
22
- HF_FILENAME = "swin_classifier_stage1_swin_large.pth"
23
  LOCAL_CKPT_DIR = "./checkpoints"
24
  MODEL_NAME = "swin_large_patch4_window12_384" # ← 使用 large
25
  NUM_CLASSES = 2
 
19
 
20
  # --- Configuration ---------------------------------------------------------
21
  REPO_ID = "telecomadm1145/swin-ai-detection"
22
+ HF_FILENAME = "swin_classifier_stage1_v2_epoch_3.pth"
23
  LOCAL_CKPT_DIR = "./checkpoints"
24
  MODEL_NAME = "swin_large_patch4_window12_384" # ← 使用 large
25
  NUM_CLASSES = 2