Update app.py

app.py

@@ -38,11 +38,8 @@ from mask_former_trainer import setup, Trainer
 
 def load_model_cfg(dataset=None):
 
-    args = SimpleNamespace(config_file='configs/maskformer/maskformer_R50_bs16_160k_dino.yaml', opts=["GWM.DATASET", dataset], wandb_sweep_mode=False, resume_path=str('checkpoints/checkpoint_best.pth'), eval_only=True)
+    args = SimpleNamespace(config_file='configs/maskformer/maskformer_R50_bs16_160k_dino.yaml', opts=["MODEL.DEVICE", "cpu", "GWM.DATASET", dataset], wandb_sweep_mode=False, resume_path=str('checkpoints/checkpoint_best.pth'), eval_only=True)
     cfg = setup(args)
-    cfg.defrost()
-    cfg.MODEL.DEVICE = 'cpu'
-    cfg.freeze()
     random_state = ut.random_state.PytorchRNGState(seed=cfg.SEED).to(torch.device(cfg.MODEL.DEVICE))
 
     model = Trainer.build_model(cfg)