Update app.py

app.py

@@ -13,7 +13,7 @@ from transformers import CLIPTextModel, CLIPTokenizer
 # from annotator.midas import MidasDetector
 # import sys
 # sys.path.insert(0, ".")
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, snapshot_download
 import controlnet_aux
 from controlnet_aux import OpenposeDetector, CannyDetector, MidasDetector
 from controlnet_aux.open_pose.body import Body
@@ -27,6 +27,7 @@ from models.RIFE.IFNet_HDv3 import IFNet
 hf_token = os.environ.get('HF_TOKEN')
 device = "cuda"
 
+snapshot_download("runwayml/stable-diffusion-v1-5", cache_dir="checkpoints")
 sd_path = "checkpoints/stable-diffusion-v1-5"
 inter_path = "checkpoints/flownet.pkl"
 controlnet_dict = {