Spaces:
Running
Running
Test Cuda
Browse files- funciones.py +1 -1
funciones.py
CHANGED
@@ -82,7 +82,7 @@ def mass(input1, input2):
|
|
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 o azure?
|
84 |
#(choose from 'tensorrt', 'cuda', 'cpu')
|
85 |
-
command = f"python run.py -s {source_path} -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider
|
86 |
print(command)
|
87 |
time.sleep(1)
|
88 |
proc = os.popen(command)
|
|
|
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 o azure?
|
84 |
#(choose from 'tensorrt', 'cuda', 'cpu')
|
85 |
+
command = f"python run.py -s {source_path} -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider cuda"
|
86 |
print(command)
|
87 |
time.sleep(1)
|
88 |
proc = os.popen(command)
|