Update face_enhancer.py

face_enhancer.py

@@ -5,17 +5,21 @@ import gfpgan
 from PIL import Image
 from upscaler.RealESRGAN import RealESRGAN
 from upscaler.codeformer import CodeFormerEnhancer
+import spaces
 
+@spaces.GPU(enable_queue=True)
 def gfpgan_runner(img, model):
     _, imgs, _ = model.enhance(img, paste_back=True, has_aligned=True)
     return imgs[0]
 
 
+@spaces.GPU(enable_queue=True)
 def realesrgan_runner(img, model):
     img = model.predict(img)
     return img
 
 
+@spaces.GPU(enable_queue=True)
 def codeformer_runner(img, model):
     img = model.enhance(img)
     return img
@@ -40,6 +44,7 @@ def get_available_enhancer_names():
     return available
 
 
+@spaces.GPU(enable_queue=True)
 def load_face_enhancer_model(name='GFPGAN', device='cuda' if torch.cuda.is_available() else 'cpu'):
     assert name in get_available_enhancer_names() + cv2_interpolations, f"Face enhancer {name} unavailable."
     if name in supported_enhancers.keys():