Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,6 @@ import os
|
|
3 |
import torch
|
4 |
import spaces
|
5 |
|
6 |
-
@spaces.GPU
|
7 |
if torch.cuda.is_available():
|
8 |
device="cuda"
|
9 |
print("Using GPU")
|
@@ -19,6 +18,7 @@ os.chdir("facefusion")
|
|
19 |
# installation
|
20 |
subprocess.run(["python", "install.py", "--onnxruntime", "cuda-11.8", "--skip-conda"], check=True)
|
21 |
|
|
|
22 |
# Run the ui
|
23 |
if device=="cuda":
|
24 |
subprocess.run(["python", "run.py", "--execution-providers", "cuda"], check=True)
|
|
|
3 |
import torch
|
4 |
import spaces
|
5 |
|
|
|
6 |
if torch.cuda.is_available():
|
7 |
device="cuda"
|
8 |
print("Using GPU")
|
|
|
18 |
# installation
|
19 |
subprocess.run(["python", "install.py", "--onnxruntime", "cuda-11.8", "--skip-conda"], check=True)
|
20 |
|
21 |
+
@spaces.GPU
|
22 |
# Run the ui
|
23 |
if device=="cuda":
|
24 |
subprocess.run(["python", "run.py", "--execution-providers", "cuda"], check=True)
|