Spaces:
Running
Running
Update app.py
Browse files
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
|