Update app.py

app.py

@@ -6,13 +6,13 @@ from tqdm import tqdm
 from huggingface_hub import snapshot_download
 from models import AudioDiffusion, DDPMScheduler
 from audioldm.audio.stft import TacotronSTFT
-#from audioldm.variational_autoencoder import AutoencoderKL
+from audioldm.variational_autoencoder import AutoencoderKL
 from pydub import AudioSegment
 from gradio import Markdown
 import spaces
 
 import torch
-from diffusers.models.autoencoder_kl import AutoencoderKL
+#from diffusers.models.autoencoder_kl import AutoencoderKL
 from diffusers.models.unet_2d_condition import UNet2DConditionModel
 from diffusers import DiffusionPipeline,AudioPipelineOutput
 from transformers import CLIPTextModel, T5EncoderModel, AutoModel, T5Tokenizer, T5TokenizerFast