Update app.py

app.py

@@ -19,7 +19,6 @@ if has_cuda:
 model.to(device)
 model.load_state_dict(load_checkpoint('base', device))
 print('total base parameters', sum(x.numel() for x in model.parameters()))has_cuda = th.cuda.is_available()
-device = th.device('cpu' if not has_cuda else 'cuda')
 def show_images(batch: th.Tensor):
     """ Display a batch of images inline. """
     scaled = ((batch + 1)*127.5).round().clamp(0,255).to(th.uint8).cpu()