Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ pipe = pipe.to("cuda")
|
|
11 |
|
12 |
def predict(prompt):
|
13 |
spec = pipe(prompt).images[0]
|
|
|
14 |
wav = wav_bytes_from_spectrogram_image(spec)
|
15 |
with open("output.wav", "wb") as f:
|
16 |
f.write(wav[0].getbuffer())
|
|
|
11 |
|
12 |
def predict(prompt):
|
13 |
spec = pipe(prompt).images[0]
|
14 |
+
print(spec)
|
15 |
wav = wav_bytes_from_spectrogram_image(spec)
|
16 |
with open("output.wav", "wb") as f:
|
17 |
f.write(wav[0].getbuffer())
|