alexcanton commited on
Commit
032f754
·
verified ·
1 Parent(s): 1e8d5eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -3,7 +3,6 @@ import os
3
  import torch
4
 
5
  if torch.cuda.is_available():
6
- # subprocess.run(["apt-get", "install", "-y", "cuda-toolkit"], check=True)
7
  device="cuda"
8
  print("Using GPU")
9
  else:
@@ -11,12 +10,12 @@ else:
11
  print("Using CPU")
12
 
13
  # Clone the repository
14
- subprocess.run(["git", "clone", "https://github.com/cantonalex/facefusion"], check=True)
15
  # chande directory to face fusion to run ui
16
  os.chdir("facefusion")
17
 
18
  # installation
19
- subprocess.run(["python install.py --torch cuda --onnxruntime cuda --skip-venv"], check=True)
20
 
21
  # Run the ui
22
  if device=="cuda":
 
3
  import torch
4
 
5
  if torch.cuda.is_available():
 
6
  device="cuda"
7
  print("Using GPU")
8
  else:
 
10
  print("Using CPU")
11
 
12
  # Clone the repository
13
+ subprocess.run(["git", "clone", "https://github.com/facefusion/facefusion", "--branch", "2.5.2", "--single-branch"], check=True)
14
  # chande directory to face fusion to run ui
15
  os.chdir("facefusion")
16
 
17
  # installation
18
+ subprocess.run(["python", "install.py", "--onnxruntime", "cuda-11.8", "--skip-conda"], check=True)
19
 
20
  # Run the ui
21
  if device=="cuda":