Update app.py

app.py

@@ -1,24 +1,24 @@
 import gradio as gr
-import torch
-from diffusers import AudioLDM2Pipeline
+# import torch
+# from diffusers import AudioLDM2Pipeline
 from share_btn import community_icon_html, loading_icon_html, share_js
 
 
 # make Space compatible with CPU duplicates
-if torch.cuda.is_available():
-    device = "cuda"
-    torch_dtype = torch.float16
-else:
-    device = "cpu"
-    torch_dtype = torch.float32
+# if torch.cuda.is_available():
+#     device = "cuda"
+#     torch_dtype = torch.float16
+# else:
+#     device = "cpu"
+#     torch_dtype = torch.float32
 
 # load the diffusers pipeline
-repo_id = "cvssp/audioldm2"
-pipe = AudioLDM2Pipeline.from_pretrained(repo_id, torch_dtype=torch_dtype).to(device)
-# pipe.unet = torch.compile(pipe.unet)
+# repo_id = "cvssp/audioldm2"
+# pipe = AudioLDM2Pipeline.from_pretrained(repo_id, torch_dtype=torch_dtype).to(device)
+# # pipe.unet = torch.compile(pipe.unet)
 
-# set the generator for reproducibility
-generator = torch.Generator(device)
+# # set the generator for reproducibility
+# generator = torch.Generator(device)
 
 
 def text2audio(text, negative_prompt, duration, guidance_scale, random_seed, n_candidates):