Update app.py

app.py

@@ -1,28 +1,6 @@
 import os
 os.system("pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/download.html")
 
-import gradio as gr
-import torch
-
-import numpy as np
-from PIL import Image
-import cv2
-import tempfile
-import moviepy.editor as mp
-from facexlib.utils.face_restoration_helper import FaceRestoreHelper
-from diffusers.utils import export_to_video, load_image
-
-# Import required modules from SkyReels
-from skyreels_a1.models.transformer3d import CogVideoXTransformer3DModel
-from skyreels_a1.skyreels_a1_i2v_pipeline import SkyReelsA1ImagePoseToVideoPipeline
-from skyreels_a1.pre_process_lmk3d import FaceAnimationProcessor
-from skyreels_a1.src.media_pipe.mp_utils import LMKExtractor
-from skyreels_a1.src.media_pipe.draw_util_2d import FaceMeshVisualizer2d
-
-from diffusers.models import AutoencoderKLCogVideoX
-from transformers import SiglipImageProcessor, SiglipVisionModel
-from diffposetalk.diffposetalk import DiffPoseTalk
-
 from huggingface_hub import snapshot_download
 
 os.makedirs("pretrained_models", exist_ok=True)
@@ -56,6 +34,28 @@ snapshot_download(
     allow_patterns="SkyReels-A1-5B/**"
 )
 
+import gradio as gr
+import torch
+
+import numpy as np
+from PIL import Image
+import cv2
+import tempfile
+import moviepy.editor as mp
+from facexlib.utils.face_restoration_helper import FaceRestoreHelper
+from diffusers.utils import export_to_video, load_image
+
+# Import required modules from SkyReels
+from skyreels_a1.models.transformer3d import CogVideoXTransformer3DModel
+from skyreels_a1.skyreels_a1_i2v_pipeline import SkyReelsA1ImagePoseToVideoPipeline
+from skyreels_a1.pre_process_lmk3d import FaceAnimationProcessor
+from skyreels_a1.src.media_pipe.mp_utils import LMKExtractor
+from skyreels_a1.src.media_pipe.draw_util_2d import FaceMeshVisualizer2d
+
+from diffusers.models import AutoencoderKLCogVideoX
+from transformers import SiglipImageProcessor, SiglipVisionModel
+from diffposetalk.diffposetalk import DiffPoseTalk
+
 
 # Helper functions from the original script
 def parse_video(driving_frames, max_frame_num, fps=25):