Charbel Malo commited on
Commit
0b8d6ef
1 Parent(s): afaf087

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import cv2
3
  import glob
4
  import time
@@ -129,7 +130,7 @@ load_face_swapper_model()
129
 
130
  ## ------------------------------ MAIN PROCESS ------------------------------
131
 
132
-
133
  def process(
134
  input_type,
135
  image_path,
 
1
  import os
2
+ import spaces
3
  import cv2
4
  import glob
5
  import time
 
130
 
131
  ## ------------------------------ MAIN PROCESS ------------------------------
132
 
133
+ @spaces.GPU
134
  def process(
135
  input_type,
136
  image_path,