Update app.py

app.py

@@ -568,7 +568,6 @@ class BaseModel(nn.Module):
     
     def __init__(self, num_classes, device='cuda' if torch.cuda.is_available() else 'cpu'):
         super().__init__()
-        self.device = device_mgr.get_optimal_device()
         self.backbone = efficientnet_v2_m(weights=EfficientNet_V2_M_Weights.IMAGENET1K_V1)
         self.feature_dim = self.backbone.classifier[1].in_features
         self.backbone.classifier = nn.Identity()
@@ -582,8 +581,6 @@ class BaseModel(nn.Module):
             nn.Linear(self.feature_dim, num_classes)
         )
 
-        self.to(device)
-
     def forward(self, x):
         x = x.to(self.device)
         features = self.backbone(x)