Update model.py

model.py

@@ -30,5 +30,5 @@ def choice_to_tensor(choice: Sequence[str]) -> torch.Tensor:
     return torch.tensor(instruments.index(choice))
 
 def generate(choice: Sequence[str], params: Sequence[int]=None):
-    noise = torch.tensor(params).unsqueeze(0).to(device) if params else torch.randn(1, 5).to('cuda')
+    noise = torch.tensor(params).unsqueeze(0).to(device) if params else torch.randn(1, 5).to(device)
     return model.sample(eps=noise, c = choice_to_tensor(choice).to(device)).cpu().numpy()[0]