Update app.py

app.py

@@ -9,6 +9,11 @@ from huggingface_hub import hf_hub_download
 import numpy as np
 from PIL import Image
 import random
+import random
+import logging
+import torchaudio
+import os
+import gc
 
 os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:512'
 os.environ['HF_HUB_CACHE'] = '/tmp/hub'  # Use temp directory to avoid filling persistent storage