Zero GPU readiness

app.py

@@ -7,7 +7,7 @@ import glob
 from moviepy.editor import VideoFileClip
 from datetime import datetime
 import time
-
+import spaces
 is_shared_ui = True if "fffiloni/X-Portrait" in os.environ['SPACE_ID'] else False
 
 # Ensure 'checkpoint' directory exists
@@ -92,6 +92,7 @@ def extract_frames_with_labels(video_path, base_output_dir="frames"):
     return  frame_data
 
 # Define a function to run your script with selected inputs
+@spaces.GPU(duration=250)
 def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_frames, num_mix, ddim_steps, progress=gr.Progress(track_tqdm=True)):
 
     start_time = time.perf_counter()