DanLeBossDeESGI commited on
Commit
15b3fd6
·
1 Parent(s): b3f9bcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ pipe.unet = torch.compile(pipe.unet)
18
 
19
  # CLAP model (only required for automatic scoring)
20
 
21
- clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to("cuda")
22
 
23
  processor = AutoProcessor.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full")
24
 
 
18
 
19
  # CLAP model (only required for automatic scoring)
20
 
21
+ clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)
22
 
23
  processor = AutoProcessor.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full")
24