Update app.py

app.py

@@ -18,7 +18,7 @@ if has_cuda:
     model.convert_to_fp16()
 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()
+print('total base parameters', sum(x.numel() for x in model.parameters()))
 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()