Spaces:
Runtime error
Runtime error
Test gpu on cuda
Browse files- funciones.py +1 -1
funciones.py
CHANGED
@@ -81,7 +81,7 @@ def mass(input1, input2):
|
|
81 |
print("source_path: ", source_path)
|
82 |
print("target_path: ", target_path)
|
83 |
#FUTURE: Agrega por parámetro o mejor aún por enviroment el hecho de si es compu para usar cpu o si es hf para usar cuda.
|
84 |
-
command = f"python run.py -s {source_path} -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider
|
85 |
print(command)
|
86 |
time.sleep(1)
|
87 |
proc = os.popen(command)
|
|
|
81 |
print("source_path: ", source_path)
|
82 |
print("target_path: ", target_path)
|
83 |
#FUTURE: Agrega por parámetro o mejor aún por enviroment el hecho de si es compu para usar cpu o si es hf para usar cuda.
|
84 |
+
command = f"python run.py -s {source_path} -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider cuda"
|
85 |
print(command)
|
86 |
time.sleep(1)
|
87 |
proc = os.popen(command)
|