Update app.py

app.py

@@ -1,5 +1,12 @@
 import subprocess
 import os
+import torch
+
+if torch.cuda.is_available():
+  !apt-get install nvidia-cuda-toolkit
+  print("Using GPU")
+else:
+  print("Using CPU")
 
 # Clone the repository
 subprocess.run(["git", "clone", "https://github.com/facefusion/facefusion"], check=True)
@@ -7,5 +14,8 @@ subprocess.run(["git", "clone", "https://github.com/facefusion/facefusion"], che
 # chande directory to face fusion to run ui
 os.chdir("facefusion")
 
+# installation
+subprocess.run(["python", "install.py", "--onnxruntime", "cuda-11.8", "--skip-venv"], check=True)
+
 # Run the ui
 subprocess.run(["python", "run.py", "--execution-providers", "cpu"], check=True)