Update app.py

app.py

@@ -570,7 +570,7 @@ class BaseModel(nn.Module):
     
     def __init__(self, num_classes, device='cuda' if torch.cuda.is_available() else 'cpu'):
         super().__init__()
-        self.device = self.device_mgr.get_optimal_device()
+        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()