init

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 import numpy as np
 import cv2
 import os
@@ -483,7 +484,8 @@ class Drag:
         viz_esti_flows = flow_to_image(controlnet_flow)  # [h, w, c]
 
         return viz_esti_flows
-
+    
+    @spaces.GPU(duration=500)
     def run(self, first_frame_path, tracking_points, inference_batch_size, motion_brush_mask, motion_brush_viz, ctrl_scale):
         
         original_width, original_height = self.width, self.height