use explicit code instead of relying on release download

Architectures/ControllabilityGAN/wgan/wgan_qc.py

@@ -242,6 +242,7 @@ class WassersteinGanQuadraticCost:
         else:
             latent_samples = self.G.sample_latent(num_samples, self.G.z_dim)
         latent_samples = latent_samples.to(self.device)
+        print(self.device)
         if nograd:
             with torch.no_grad():
                 generated_data = self.G(latent_samples, return_intermediate=return_intermediate)

app.py

@@ -136,6 +136,8 @@ def read(prompt,
     if torch.cuda.is_available():
         controllable_ui.to("cuda")
         controllable_ui.device = "cuda"
+        controllable_ui.model.device = "cuda"
+        controllable_ui.wgan.device = "cuda"
     try:
         sr, wav, fig = controllable_ui.read(prompt,
                                             language.split(" ")[-1].split("(")[1].split(")")[0],
@@ -155,6 +157,8 @@ def read(prompt,
     finally:
         controllable_ui.to("cpu")
         controllable_ui.device = "cpu"
+        controllable_ui.model.device = "cpu"
+        controllable_ui.wgan.device = "cpu"
     return (sr, float2pcm(wav)), fig