Update app.py

app.py

@@ -8,7 +8,7 @@ import numpy as np
 from RealESRGAN import RealESRGAN
 
 os.environ['CUDA_HOME'] = "/usr/lib/x86_64-linux-gnu"
-subprocess.run(["nvidia-smi"])
+
 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"])
 subprocess.run(["mv", "./network-refinement.pytorch", "./3d-ken-burns/models/disparity-refinement.pytorch"])
@@ -33,6 +33,7 @@ def upscale(image):
 
 @spaces.GPU
 def generate_video(image):
+    subprocess.run(["nvidia-smi"])
     image = Image.fromarray(image)
     image.save("/tmp/img.png")
     os.chdir("./3d-ken-burns")