Update app.py

app.py

@@ -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)
         )