Update app.py

app.py

@@ -150,7 +150,7 @@ class Generator(torch.nn.Module):
                 torch.nn.init.normal_(m.conv.weight,mean,std)
                 torch.nn.init.constant_(m.conv.bias,0)
 
-model = Generator(3, 32, 3, 4).cuda() # input_dim, num_filter, output_dim, num_resnet
+model = Generator(3, 32, 3, 4).cpu() # input_dim, num_filter, output_dim, num_resnet
 model.load_state_dict(torch.load('G_A_HW4_SAVE.pt',map_location=torch.device('cpu')))
 print(model)
 model.eval()