Update app.py
Browse files
app.py
CHANGED
|
@@ -7,8 +7,6 @@ from huggingface_hub import snapshot_download
|
|
| 7 |
import soundfile as sf
|
| 8 |
from auffusion_pipeline import AuffusionPipeline
|
| 9 |
|
| 10 |
-
pipeline = AuffusionPipeline.from_pretrained("auffusion/auffusion")
|
| 11 |
-
|
| 12 |
# ——
|
| 13 |
|
| 14 |
from diffusers import StableDiffusionImg2ImgPipeline
|
|
@@ -26,7 +24,7 @@ def save_spectrogram_image(spectrogram, filename):
|
|
| 26 |
plt.close()
|
| 27 |
|
| 28 |
def infer(prompt, progress=gr.Progress(track_tqdm=True)):
|
| 29 |
-
|
| 30 |
prompt = prompt
|
| 31 |
output = pipeline(prompt=prompt)
|
| 32 |
audio = output.audios[0]
|
|
|
|
| 7 |
import soundfile as sf
|
| 8 |
from auffusion_pipeline import AuffusionPipeline
|
| 9 |
|
|
|
|
|
|
|
| 10 |
# ——
|
| 11 |
|
| 12 |
from diffusers import StableDiffusionImg2ImgPipeline
|
|
|
|
| 24 |
plt.close()
|
| 25 |
|
| 26 |
def infer(prompt, progress=gr.Progress(track_tqdm=True)):
|
| 27 |
+
pipeline = AuffusionPipeline.from_pretrained("auffusion/auffusion")
|
| 28 |
prompt = prompt
|
| 29 |
output = pipeline(prompt=prompt)
|
| 30 |
audio = output.audios[0]
|