Update app.py

app.py

@@ -598,7 +598,7 @@ device_mgr = DeviceManager()
 num_classes = len(dog_breeds)
 
 # Initialize base model
-model = BaseModel(num_classes=num_classes, device=device).to(device)
+model = BaseModel(num_classes=num_classes)
 
 # Load model path
 model_path = '124_best_model_dog.pth'