Update app.py

app.py

@@ -216,6 +216,9 @@ def generate_multi_view_images(prompt, seed):
 # 重建 3D 模型
 @spaces.GPU
 def reconstruct_3d_model(images, prompt):
+    global model
+    model.init_flexicubes_geometry(device_1, fovy=50.0)
+    model = model.eval()
     rgb_normal_grid = images
     save_dir_path = os.path.join(save_dir, prompt.replace(" ", "_"))
     os.makedirs(save_dir_path, exist_ok=True)
@@ -283,9 +286,6 @@ def reconstruct_3d_model(images, prompt):
 # Gradio 接口函数
 @spaces.GPU
 def gradio_pipeline(prompt, seed):
-    global model
-    model.init_flexicubes_geometry(device_1, fovy=50.0)
-    model = model.eval()
     # 生成多视图图像
     rgb_normal_grid = generate_multi_view_images(prompt, seed)
     image_preview = Image.fromarray((rgb_normal_grid * 255).astype(np.uint8))