Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import subprocess
 import os
 import torch
@@ -9,16 +10,30 @@ else:
   device="cpu"
   print("Using CPU")
 
+
+# get branch
+git_branch = os.getenv('git_branch')
+
+
 # Clone the repository
-subprocess.run(["git", "clone", "https://github.com/facefusion/facefusion", "--branch", "2.6.1", "--single-branch"], check=True)
+subprocess.run(["git", "clone", "https://github.com/facefusion/facefusion", "--branch", git_branch, "--single-branch"], check=True)
 # chande directory to face fusion to run ui
 os.chdir("facefusion")
 
+
 # installation
-subprocess.run(["python", "install.py", "--onnxruntime", "cuda-11.8", "--skip-conda"], check=True)
+if device == "cuda":
+    subprocess.run(["python", "install.py", "--onnxruntime", "cuda-11.8", "--skip-conda"], check=True)
+elif device == "cpu":
+    subprocess.run(["python", "install.py", "--onnxruntime", "default", "--skip-conda"], check=True)
+
 
 # Run the ui
-if device=="cuda":
+if device == "cuda" and git_branch == "master":
     subprocess.run(["python", "run.py", "--execution-providers", "cuda"], check=True)
-else:
-    subprocess.run(["python", "run.py", "--execution-providers", "cpu"], check=True)
+elif device == "cpu" and git_branch == "master":
+    subprocess.run(["python", "run.py", "--execution-providers", "cpu"], check=True)
+elif device == "cuda" and git_branch == "next":
+    subprocess.run(["python", "facefusion.py", "run", "--execution-providers", "cuda"], check=True)
+elif device == "cpu" and git_branch == "next":
+    subprocess.run(["python", "facefusion.py", "run", "--execution-providers", "cpu"], check=True)