zeroGPU ready

app.py

@@ -11,6 +11,7 @@ Date: 2025/01/15
 """
 
 import torch
+import spaces
 import sys
 import os
 import subprocess
@@ -397,6 +398,7 @@ def get_overlap_slide_window_indices(video_length, window_size, window_overlap):
 def random_seed():
     return random.randint(0, 10000)
 
+@spaces.GPU(duration=180)
 def infer(lq_sequence, task_name, mask, seed, progress=gr.Progress(track_tqdm=True)):
     
     if is_shared_ui: