Spaces:
Running
on
Zero
Running
on
Zero
Stanislaw Szymanowicz
commited on
Commit
•
3d70c12
1
Parent(s):
de8d36c
Add cuda python
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -30,7 +30,7 @@ from huggingface_hub import hf_hub_download
|
|
| 30 |
|
| 31 |
def main():
|
| 32 |
|
| 33 |
-
os.system("find /usr -name nvcc")
|
| 34 |
|
| 35 |
if torch.cuda.is_available():
|
| 36 |
device = "cuda:0"
|
|
|
|
| 30 |
|
| 31 |
def main():
|
| 32 |
|
| 33 |
+
os.system("sudo find /usr -name nvcc")
|
| 34 |
|
| 35 |
if torch.cuda.is_available():
|
| 36 |
device = "cuda:0"
|
requirements.txt
CHANGED
|
@@ -8,4 +8,5 @@ moviepy
|
|
| 8 |
markupsafe==2.0.1
|
| 9 |
gradio==4.25.0
|
| 10 |
rembg
|
| 11 |
-
spaces
|
|
|
|
|
|
| 8 |
markupsafe==2.0.1
|
| 9 |
gradio==4.25.0
|
| 10 |
rembg
|
| 11 |
+
spaces
|
| 12 |
+
cuda-python
|