envs

app.py

@@ -6,6 +6,7 @@ import uuid
 import torch
 import torchvision
 import json
+import spaces
 
 from PIL import Image
 from omegaconf import OmegaConf
@@ -251,6 +252,7 @@ class ImageConductor:
 
         self.blur_kernel = blur_kernel
 
+    @spaces.GPU(enable_queue=True, duration=200)
     @torch.no_grad()
     def run(self, first_frame_path, tracking_points, prompt, drag_mode, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, personalized):