Added Cuda Features

app.py

@@ -115,7 +115,7 @@ def greet(input1, input2):
     print("source_path: ", source_path)
     print("target_path: ", target_path)
 
-    command = f"python run.py -s {source_path}  -t {target_path} -o {result_path} --frame-processor {procesador}"
+    command = f"python run.py -s {source_path}  -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider cuda"
     print(command)
     time.sleep(1)
     proc = os.popen(command)

requirements.txt

@@ -6,9 +6,10 @@ psutil==5.9.5
 tk==0.1.0
 customtkinter==5.2.0
 tkinterdnd2==0.3.0
-onnxruntime==1.15.0
+#onnxruntime==1.15.0
 tensorflow==2.13.0
 opennsfw2==0.10.2
 protobuf==4.23.4
 tqdm==4.65.0
-zipfile36
+zipfile36
+onnxruntime-gpu==1.15.1