meepmoo commited on
Commit
ba7db10
·
verified ·
1 Parent(s): 5b336be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)