alphadead commited on
Commit
0e9a5a0
·
verified ·
1 Parent(s): f89d1dc

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import numpy as np
2
  import gradio as gr
3
  import roop.globals
@@ -12,7 +13,7 @@ from roop.utilities import normalize_output_path
12
  import os
13
  from PIL import Image
14
 
15
-
16
  def swap_face(source_file, target_file, doFaceEnhancer):
17
 
18
  source_path = "input.jpg"
 
1
+ import spaces
2
  import numpy as np
3
  import gradio as gr
4
  import roop.globals
 
13
  import os
14
  from PIL import Image
15
 
16
+ @spaces.GPU(duration=120)
17
  def swap_face(source_file, target_file, doFaceEnhancer):
18
 
19
  source_path = "input.jpg"