load model

app.py

@@ -5,6 +5,7 @@ import demo_tasks
 from functools import partial
 from data.prefix_instruction import get_layout_instruction
 from visualcloze import VisualClozeModel
+from huggingface_hub import snapshot_download
 
 
 max_grid_h = 5
@@ -469,6 +470,8 @@ def parse_args():
 if __name__ == "__main__":
     args = parse_args()
     
+    snapshot_download(repo_id="VisualCloze/VisualCloze", repo_type="model", local_dir="models")
+
     # Initialize model
     model = VisualClozeModel(resolution=args.resolution, model_path=args.model_path, precision=args.precision)