Update app.py

app.py

@@ -7,7 +7,7 @@ from PIL import Image
 import numpy as np
 from RealESRGAN import RealESRGAN
 
-os.environ['CUDA_HOME'] = "/temp/cuda"
+os.environ['CUDA_HOME'] = "user/lib/cuda"
 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', '')
 
@@ -40,7 +40,8 @@ def upscale(image):
 def generate_video(image):
     subprocess.run(["nvidia-smi"])
     subprocess.run(["find", "/", "-name", "cuda"])
-    subprocess.run(["ls","-al","/user/temp"])
+    subprocess.run(["ls","-al","/user"])
+    subprocess.run(["ls","-al","/user/local"])
     image = Image.fromarray(image)
     image.save("/tmp/img.png")
     os.chdir("./3d-ken-burns")