Update app.py

app.py

@@ -11,6 +11,14 @@ subprocess.run(
     shell=True,
 )
 
+os.makedirs("/home/user/app/checkpoints", exist_ok=True)
+from huggingface_hub import snapshot_download
+snapshot_download(
+    repo_id="Alpha-VLLM/Lumina-Image-2.0", local_dir="/home/user/app/checkpoints"
+)
+
+hf_token = os.environ["HF_TOKEN"]
+
 
 import argparse
 import os
@@ -361,7 +369,7 @@ def main():
     parser = argparse.ArgumentParser()
 
     parser.add_argument("--num_gpus", type=int, default=1)
-    parser.add_argument("--ckpt", type=str,default='/home/ubuntu/zl/T2I/ckt_256_final/', required=False)
+    parser.add_argument("--ckpt", type=str,default='/home/user/app/checkpoints', required=False)
     parser.add_argument("--ema", action="store_true")
     parser.add_argument("--precision", default="bf16", choices=["bf16", "fp32"])
     parser.add_argument("--hf_token", type=str, default=None, help="huggingface read token for accessing gated repo.")