Update app.py

app.py

@@ -22,19 +22,10 @@ import pickle
 from tqdm import tqdm
 import copy
 from argparse import Namespace
-
-from musetalk.utils.utils import get_file_type,get_video_fps,datagen
-from musetalk.utils.preprocessing import get_landmark_and_bbox,read_imgs,coord_placeholder
-from musetalk.utils.blending import get_image
-from musetalk.utils.utils import load_all_model
 import shutil
 import gdown
 
 
-
-ProjectDir = os.path.abspath(os.path.dirname(__file__))
-CheckpointsDir = os.path.join(ProjectDir, "checkpoints")
-
 def download_model():
     if not os.path.exists(CheckpointsDir):
         os.makedirs(CheckpointsDir)
@@ -98,6 +89,22 @@ def download_model():
     else:
         print("Already download the model.")
 
+
+
+download_model()  # for huggingface deployment.
+
+
+from musetalk.utils.utils import get_file_type,get_video_fps,datagen
+from musetalk.utils.preprocessing import get_landmark_and_bbox,read_imgs,coord_placeholder
+from musetalk.utils.blending import get_image
+from musetalk.utils.utils import load_all_model
+
+
+
+ProjectDir = os.path.abspath(os.path.dirname(__file__))
+CheckpointsDir = os.path.join(ProjectDir, "checkpoints")
+
+
 @spaces.GPU(duration=600)
 @torch.no_grad()
 def inference(audio_path,video_path,bbox_shift,progress=gr.Progress(track_tqdm=True)):
@@ -203,7 +210,6 @@ def inference(audio_path,video_path,bbox_shift,progress=gr.Progress(track_tqdm=T
     print(f"result is save to {output_vid_name}")
     return output_vid_name
 
-download_model()  # for huggingface deployment.
 
 
 # load model weights