Update app.py

app.py

@@ -24,15 +24,16 @@ from gradio_demo.utils_drag import *
 import warnings
 print("gr file", gr.__file__)
 
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, snapshot_download
 os.makedirs("checkpoints", exist_ok=True)
-hf_hub_download(
+snapshot_download(
     "wwen1997/framer_512x320",
     local_dir="checkpoints/framer_512x320",
     token=os.environ["TOKEN"],
 )
 
 
+
 def get_args():
     import argparse
     parser = argparse.ArgumentParser()