MaxGab commited on
Commit
e89ad7d
·
verified ·
1 Parent(s): 4b839e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,8 +1,14 @@
1
  from diffusers import AudioLDM2Pipeline
 
2
  import torch
3
 
4
  model = "cvssp/audioldm2"
5
-
6
  pipe = AudioLDM2Pipeline.from_pretrained(model, torch_dtype = torch.float16,).to("cuda")
7
 
8
- #https://www.google.com/search?sca_esv=cd968c0da757ee90&q=how+to+integrate+facebook+small+music+gen+ai+model+in+hugging+face&udm=7&fbs=ABzOT_CWdhQLP1FcmU5B0fn3xuWp6IcynRBrzjy_vjxR0KoDMp_4ut2Z3jppK72fzdIpWsBpYmR8fwcVczrRGmP-Hf4k8E3HhH0FkewPslVYtPbS7rGVU8bggt4DoRlXFprvjJNrWGq5bwD453K6Qh-s0BRNe1W_hAxmbVQyPGbJvui05wbS7HVKzL9e-c2LYt6maQuIGup0S0PHSh7uopV_cX6MhI6o3A&sa=X&ved=2ahUKEwiV07K16ZiMAxXbSzABHVsFKcwQtKgLegQIExAB&biw=1512&bih=787&dpr=2#fpstate=ive&vld=cid:21b04632,vid:_5F-toek0G8,st:0
 
 
 
 
 
 
 
1
  from diffusers import AudioLDM2Pipeline
2
+ from IPython.display import Audio
3
  import torch
4
 
5
  model = "cvssp/audioldm2"
 
6
  pipe = AudioLDM2Pipeline.from_pretrained(model, torch_dtype = torch.float16,).to("cuda")
7
 
8
+ prompt = "A cheerful ukulele strumming in a beachside jam."
9
+
10
+ generator = torch.Generator('cuda').manual_seed(0)
11
+
12
+ audio = pipe(prompt,audio_length_in_s = 10.24, generator = generator).audios[0]
13
+
14
+ Audio(audio, rate = 16000)