Moved init models

app.py

@@ -226,23 +226,12 @@ class Drag:
         self.device = device
 
         ckpts_dir = "./ckpts/"
-        svd_ckpt = "./ckpts/stable-video-diffusion-img2vid-xt-1-1"
-        mofa_ckpt = "./ckpts/controlnet/ckpts/controlnet"
 
         self.device = 'cuda'
         self.weight_dtype = torch.float16
         
         download_models(ckpts_dir)
 
-        print("Contents of ckpts_dir:", os.listdir(f"{ckpts_dir}/controlnet"))
-
-        self.pipeline, self.cmp = init_models(
-            svd_ckpt, 
-            mofa_ckpt, 
-            weight_dtype=self.weight_dtype, 
-            device=self.device
-        )
-
         self.height = height
         self.width = width
         self.model_length = model_length
@@ -500,6 +489,16 @@ class Drag:
 
     @spaces.GPU(enable_queue=True, duration=240)
     def run(self, first_frame_path, tracking_points, inference_batch_size, motion_brush_mask, motion_brush_viz, ctrl_scale):
+        svd_ckpt = "./ckpts/stable-video-diffusion-img2vid-xt-1-1"
+        mofa_ckpt = "./ckpts/controlnet/ckpts/controlnet"
+    
+        self.pipeline, self.cmp = init_models(
+            svd_ckpt, 
+            mofa_ckpt, 
+            weight_dtype=self.weight_dtype, 
+            device=self.device
+        )
+
         self.pipeline = self.pipeline.to("cuda:0")
         self.cmp = self.cmp.to("cuda:0")