Added device variable

app.py

@@ -7,15 +7,17 @@ from transformers import AutoProcessor, ClapModel
 import torch
 # import scipy
 
+device="cpu"
+
 repo_id = "cvssp/audioldm-s-full-v2"
 pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
 #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-pipe = pipe.to("cpu")
+pipe = pipe.to(device)
 
 clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)
 processor = AutoProcessor.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full")
 
-generator = torch.Generator("cpu")
+generator = torch.Generator(device)
 
 def texttoaudio(prompt, neg_prompt, seed, inf_steps, guidance_scale):
     if prompt is None: