Update app.py

app.py

@@ -49,7 +49,6 @@ for key, value in checkpoint['model'].items():
 load_result = model.load_state_dict(new_ckpt, strict=False)
 assert len(load_result.unexpected_keys) == 0, f"Unexpected keys: {load_result.unexpected_keys}"
 model.eval()
-model.to("cuda")
 
 transform = transforms.Compose(
     [transforms.ToTensor(), transforms.Lambda(lambda x: x.repeat(3, 1, 1) if x.size(0) == 1 else x)])