Try829 commited on
Commit
2c6812e
·
verified ·
1 Parent(s): 9415dd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import os
2
  import cv2
3
  import glob
4
  import time
@@ -8,6 +8,8 @@ import argparse
8
  import platform
9
  import datetime
10
  import subprocess
 
 
11
  import insightface
12
  import onnxruntime
13
  import numpy as np
 
1
+
2
  import cv2
3
  import glob
4
  import time
 
8
  import platform
9
  import datetime
10
  import subprocess
11
+ import os
12
+ os.environ["INSIGHTFACE_USE_GPU"] = "0" # force CPU mode
13
  import insightface
14
  import onnxruntime
15
  import numpy as np