Update app.py

app.py

@@ -17,11 +17,21 @@ from utils.pipeline_magictime import MagicTimePipeline
 from utils.util import save_videos_grid, convert_ldm_unet_checkpoint, convert_ldm_clip_checkpoint, convert_ldm_vae_checkpoint, load_diffusers_lora_unet, convert_ldm_clip_text_model
 import spaces
 
-pretrained_model_path   = "./ckpts/Base_Model/stable-diffusion-v1-5"
-inference_config_path   = "./sample_configs/RealisticVision.yaml"
-magic_adapter_s_path    = "./ckpts/Magic_Weights/magic_adapter_s/magic_adapter_s.ckpt"
-magic_adapter_t_path    = "./ckpts/Magic_Weights/magic_adapter_t"
-magic_text_encoder_path = "./ckpts/Magic_Weights/magic_text_encoder"
+from huggingface_hub import hf_hub_download, snapshot_download
+
+model_path = "ckpts"
+
+if not os.path.exists(model_path) or not os.path.exists(f"{model_path}/model_real_esran") or not os.path.exists(f"{model_path}/model_rife"):
+    print("Model not found, downloading from Hugging Face...")
+    snapshot_download(repo_id="BestWishYsh/MagicTime", local_dir=f"{model_path}")
+else:
+    print(f"Model already exists in {model_path}, skipping download.")
+    
+pretrained_model_path   = "ckpts/Base_Model/stable-diffusion-v1-5"
+inference_config_path   = "sample_configs/RealisticVision.yaml"
+magic_adapter_s_path    = "ckpts/Magic_Weights/magic_adapter_s/magic_adapter_s.ckpt"
+magic_adapter_t_path    = "ckpts/Magic_Weights/magic_adapter_t"
+magic_text_encoder_path = "ckpts/Magic_Weights/magic_text_encoder"
 
 css = """
 .toolbutton {