Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|