From d009b0a394f44445dc295e9f13a3797dcbeecf06 Mon Sep 17 00:00:00 2001
From: AIFSH <149982694+AIFSH@users.noreply.github.com>
Date: Sat, 1 Jun 2024 10:47:27 +0800
Subject: [PATCH] Update inference.py

---
 V_Express/inference.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/V_Express/inference.py b/V_Express/inference.py
index 5354b89..57d320d 100755
--- a/V_Express/inference.py
+++ b/V_Express/inference.py
@@ -1,6 +1,6 @@
 import argparse
 
-import os
+import os,sys
 import cv2
 import numpy as np
 import torch
@@ -184,8 +184,8 @@ def main():
     ).to(dtype=dtype, device=device)
 
     app = FaceAnalysis(
-        providers=['CUDAExecutionProvider' if args.device == 'cuda' else 'CPUExecutionProvider'],
-        provider_options=[{'device_id': args.gpu_id}] if args.device == 'cuda' else [],
+        providers=['CUDAExecutionProvider' if args.device == 'cuda' and sys.platform != "win32" else 'CPUExecutionProvider'],
+        provider_options=[{'device_id': args.gpu_id}] if args.device == 'cuda' and sys.platform != "win32" else [],
         root=args.insightface_model_path,
     )
     app.prepare(ctx_id=0, det_size=(args.image_height, args.image_width))