Update app.py

app.py

@@ -17,7 +17,9 @@ pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype).to(dev
 pipe.unet = torch.compile(pipe.unet)
 
 # CLAP model (only required for automatic scoring)
-clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)
+
+clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to("cuda")
+
 processor = AutoProcessor.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full")
 
 generator = torch.Generator(device)