From c18f66697aef73617e232e4b236fdddc746a8b98 Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Mon, 29 May 2023 17:39:29 +0530 Subject: [PATCH] fetch available providers --- core/config.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/config.py b/core/config.py index 1edf36c..fb69913 100644 --- a/core/config.py +++ b/core/config.py @@ -1,6 +1,7 @@ import insightface +import onnxruntime -face_analyser = insightface.app.FaceAnalysis(name='buffalo_l', providers=['CUDAExecutionProvider', 'ROCMExecutionProvider', 'CPUExecutionProvider']) +face_analyser = insightface.app.FaceAnalysis(name='buffalo_l', providers=onnxruntime.get_available_providers()) face_analyser.prepare(ctx_id=0, det_size=(640, 640))