update

app.py

@@ -10,6 +10,7 @@ from sixdrepnet.model import SixDRepNet
 import pixel_generator.vec2face.model_vec2face as model_vec2face
 MAX_SEED = np.iinfo(np.int32).max
 import torch
+import spaces
 from time import time
 
 
@@ -79,7 +80,7 @@ def initialize_models():
 
     return generator, id_model, pose_model, quality_model
 
-
+@spaces.GPU
 def image_generation(input_image, quality, use_target_pose, pose, dimension, progress=gr.Progress()):
     generator, id_model, pose_model, quality_model = initialize_models()
 
@@ -120,7 +121,7 @@ def image_generation(input_image, quality, use_target_pose, pose, dimension, pro
 
     return generated_images
 
-
+@spaces.GPU
 def process_input(image_input, num1, num2, num3, num4, random_seed, target_quality, use_target_pose, target_pose, progress=gr.Progress()):
     # Ensure all dimension numbers are within [0, 512)
     num1, num2, num3, num4 = [max(0, min(int(n), 511)) for n in [num1, num2, num3, num4]]