Gregniuki commited on
Commit
5b0c0b4
·
1 Parent(s): 4cc1d5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -238,6 +238,7 @@ async def main(
238
  if use_gpu is False
239
  else ("CUDAExecutionProvider", {"cudnn_conv_algo_search": "DEFAULT"})
240
  ]
 
241
  sess_options = onnxruntime.SessionOptions()
242
  model, config = load_onnx(selected_model, sess_options, providers)
243
  config["espeak"]["voice"] = speaker
 
238
  if use_gpu is False
239
  else ("CUDAExecutionProvider", {"cudnn_conv_algo_search": "DEFAULT"})
240
  ]
241
+ print(onnxruntime.get_device())
242
  sess_options = onnxruntime.SessionOptions()
243
  model, config = load_onnx(selected_model, sess_options, providers)
244
  config["espeak"]["voice"] = speaker