Change code for huggingface

app.py

@@ -38,7 +38,7 @@ beta1 = 0.5
 ngpu = 0
 
 
-device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
+device = torch.device("cuda:0" if (ngpu > 0 and torch.cuda.is_available()) else "cpu")
 
 
 # custom weights initialization called on netG and netD
@@ -86,7 +86,7 @@ class Generator(nn.Module):
 netG = Generator(ngpu).to(device)
 
 # Handle multi-gpu if desired
-if (device.type == 'cuda') and (ngpu > 1):
+if (ngpu > 1) and (device.type == 'cuda'):
     netG = nn.DataParallel(netG, list(range(ngpu)))
 
 # Apply the weights_init function to randomly initialize all weights