Update app.py

app.py

@@ -4,6 +4,14 @@ import os
 import shutil
 import tempfile
 
+# Set the PATH and LD_LIBRARY_PATH for CUDA 12.3
+cuda_bin_path = "/usr/local/cuda-12.3/bin"
+cuda_lib_path = "/usr/local/cuda-12.3/lib64"
+
+# Update the environment variables
+os.environ['PATH'] = f"{cuda_bin_path}:{os.environ.get('PATH', '')}"
+os.environ['LD_LIBRARY_PATH'] = f"{cuda_lib_path}:{os.environ.get('LD_LIBRARY_PATH', '')}"
+
 from huggingface_hub import snapshot_download 
 
 # Create xcodec_mini_infer folder