talalif commited on
Commit
633db96
·
verified ·
1 Parent(s): bea517c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -39,6 +39,7 @@ def upscale(image):
39
  @spaces.GPU
40
  def generate_video(image):
41
  subprocess.run(["nvidia-smi"])
 
42
  subprocess.run(["find", "/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib", "-name", "libnvrtc*"])
43
  image = Image.fromarray(image)
44
  image.save("/tmp/img.png")
 
39
  @spaces.GPU
40
  def generate_video(image):
41
  subprocess.run(["nvidia-smi"])
42
+ subprocess.run(["locate","cuda"])
43
  subprocess.run(["find", "/usr/local/lib/python3.10/site-packages/nvidia/cuda_nvrtc/lib", "-name", "libnvrtc*"])
44
  image = Image.fromarray(image)
45
  image.save("/tmp/img.png")