Update app.py

app.py

@@ -8,6 +8,8 @@ import numpy as np
 from RealESRGAN import RealESRGAN
 
 os.environ['CUDA_HOME'] = "/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib"
+os.environ['PATH'] = '/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib:' + os.environ.get('PATH', '')
+os.environ['LD_LIBRARY_PATH'] = '/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib:' + os.environ.get('LD_LIBRARY_PATH', '')
 
 subprocess.run(["git", "clone", "https://github.com/sniklaus/3d-ken-burns.git"])
 subprocess.run(["mv", "./network-disparity.pytorch", "./3d-ken-burns/models/disparity-estimation.pytorch"])