Moibe commited on
Commit
dd615e1
·
1 Parent(s): 98a389a
Files changed (1) hide show
  1. 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 azure"
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)