Update app.py

app.py

@@ -83,7 +83,7 @@ from huggingface_hub import hf_hub_download
 
 from utils.tool import NormalTransfer, get_background, get_render_cameras_video, load_mipmap, render_frames
 
-device_0 = "cuda:1"
+device_0 = "cuda"
 device_1 = "cuda"
 resolution = 512
 save_dir = "./outputs"
@@ -222,9 +222,9 @@ def generate_multi_view_images(prompt, seed):
         for img in flux_pipe.flux_pipe_call_that_returns_an_iterable_of_images(
             prompt=prompt,
             guidance_scale=3.5,
-            num_inference_steps=10,
-            width=resolution * 2,
-            height=resolution * 1,
+            num_inference_steps=4,
+            width=resolution * 4,
+            height=resolution * 2,
             generator=generator,
             output_type="np",
             good_vae=good_vae,
@@ -307,16 +307,16 @@ def reconstruct_3d_model(images, prompt):
 @spaces.GPU
 def gradio_pipeline(prompt, seed):
     # 生成多视图图像
-    # rgb_normal_grid = generate_multi_view_images(prompt, seed)
-    rgb_normal_grid = np.load("rgb_normal_grid.npy")
+    rgb_normal_grid = generate_multi_view_images(prompt, seed)
+    # rgb_normal_grid = np.load("rgb_normal_grid.npy")
     image_preview = Image.fromarray((rgb_normal_grid[0] * 255).astype(np.uint8))
 
     # 3d reconstruction
 
 
     # 重建 3D 模型并返回 glb 路径
-    save_glb_addr = reconstruct_3d_model(rgb_normal_grid, prompt)
-
+    # save_glb_addr = reconstruct_3d_model(rgb_normal_grid, prompt)
+    save_glb_addr = None
     return image_preview, save_glb_addr
 
 # Gradio Blocks 应用