Update app.py

app.py

@@ -21,7 +21,7 @@ from pytorch_grad_cam.utils.image import show_cam_on_image
 REPO_ID          = "telecomadm1145/swin-ai-detection"
 HF_FILENAME      = "swin_classifier_stage1_swin_large.pth"
 LOCAL_CKPT_DIR   = "./checkpoints"
-MODEL_NAME       = "swin_large_patch4_window7_224"     # ← 使用 large
+MODEL_NAME       = "swin_large_patch4_window12_384"     # ← 使用 large
 NUM_CLASSES      = 2
 SEED             = 4421
 dropout_rate     = 0.1
@@ -42,15 +42,15 @@ class SwinClassifier(nn.Module):
         self.data_config = timm.data.resolve_data_config({}, model=self.backbone)
 
         self.classifier = nn.Sequential(
-            #nn.Dropout(dropout_rate),
+            nn.Dropout(dropout_rate),
             nn.Linear(self.backbone.num_features, 512),
             nn.BatchNorm1d(512),
             nn.ReLU(),
-            #nn.Dropout(dropout_rate * 0.7),
+            nn.Dropout(dropout_rate * 0.7),
             nn.Linear(512, 128),
             nn.BatchNorm1d(128),
             nn.ReLU(),
-            #nn.Dropout(dropout_rate * 0.5),
+            nn.Dropout(dropout_rate * 0.5),
             nn.Linear(128, num_classes)
         )