tuan2308 commited on
Commit
ff1d994
1 Parent(s): 28ae202

Update roop/face_analyser.py

Browse files
Files changed (1) hide show
  1. roop/face_analyser.py +2 -1
roop/face_analyser.py CHANGED
@@ -2,6 +2,7 @@ import threading
2
  from typing import Any, Optional, List
3
  import insightface
4
  import numpy
 
5
 
6
  import roop.globals
7
  from roop.typing import Frame, Face
@@ -9,7 +10,7 @@ from roop.typing import Frame, Face
9
  FACE_ANALYSER = None
10
  THREAD_LOCK = threading.Lock()
11
 
12
-
13
  def get_face_analyser() -> Any:
14
  global FACE_ANALYSER
15
 
 
2
  from typing import Any, Optional, List
3
  import insightface
4
  import numpy
5
+ import spaces
6
 
7
  import roop.globals
8
  from roop.typing import Frame, Face
 
10
  FACE_ANALYSER = None
11
  THREAD_LOCK = threading.Lock()
12
 
13
+ @spaces.GPU
14
  def get_face_analyser() -> Any:
15
  global FACE_ANALYSER
16