update

infer_api.py

@@ -100,6 +100,12 @@ weight_dtype = torch.float16
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 VIEWS = ['front', 'front_right', 'right', 'back', 'left', 'front_left']
 
+repo_id = "hyz317/StdGEN"
+all_files = list_repo_files(repo_id, revision="main")
+for file in all_files:
+    if os.path.exists(file):
+        continue
+    hf_hub_download(repo_id, file, local_dir="./ckpt")
 
 @spaces.GPU
 def set_seed(seed):
@@ -802,13 +808,6 @@ def infer_multiview_gen(img, seed, num_levels):
         all_results[k] = results[k]
     return results, all_results
 
-repo_id = "hyz317/StdGEN"
-all_files = list_repo_files(repo_id, revision="main")
-for file in all_files:
-    if os.path.exists(file):
-        continue
-    hf_hub_download(repo_id, file, local_dir="./ckpt")
-
 infer_canonicalize_config = {
     'config_path': './configs/canonicalization-infer.yaml',
 }