Update app.py

app.py

@@ -7,8 +7,8 @@ import spaces
 import gradio as gr
 import numpy as np
 import torch
-from gradio_image_prompter import ImagePrompter
-from gradio_litmodel3d import LitModel3D
+# from gradio_image_prompter import ImagePrompter
+# from gradio_litmodel3d import LitModel3D
 from huggingface_hub import snapshot_download
 from PIL import Image
 import trimesh
@@ -30,7 +30,7 @@ REPO_ID = "VAST-AI/DetailGen3D" # 似乎还没有
 
 MARKDOWN = """
 ## Generating geometry details guided by reference image with [DetailGen3D](https://detailgen3d.github.io/DetailGen3D/)
-1. Upload a detailed image of the frontal view and a coarse model. Then clik "Run " to generate the refined result. 
+1. Upload a detailed image of the frontal view and a coarse model. Then clik "Generate Details" to generate the refined result. 
 2. If you find the generated 3D scene satisfactory, download it by clicking the "Download GLB" button.
 3. If you want the refine result to be more consistent with the image, please manually increase the CFG strength.
 """
@@ -40,11 +40,38 @@ EXAMPLES = [
         "assets/model/100.glb",
         42,
         False,
-    ]
+    ],
+    [
+        "assets/image/503d193a-1b9b-4685-b05f-00ac82f93d7b.png",
+        "assets/model/503d193a-1b9b-4685-b05f-00ac82f93d7b.glb",
+        42,
+        False,
+    ],
+    [
+        "assets/image/34933195-9c2c-4271-8d31-a28bc5348b7a.png",
+        "assets/model/34933195-9c2c-4271-8d31-a28bc5348b7a.glb",
+        42,
+        False,
+    ],
+    [
+        "assets/image/a5d09c66-1617-465c-aec9-431f48d9a7e1.png",
+        "assets/model/a5d09c66-1617-465c-aec9-431f48d9a7e1.glb",
+        42,
+        False,
+    ],
+    [
+        "assets/image/cb7e6c4a-b4dd-483c-9789-3d4887ee7434.png",
+        "assets/model/cb7e6c4a-b4dd-483c-9789-3d4887ee7434.glb",
+        42,
+        False,
+    ],
+    [
+        "assets/image/e799e6b4-3b47-40e0-befb-b156af8758ad.png",
+        "assets/model/instant3d/e799e6b4-3b47-40e0-befb-b156af8758ad.glb",
+        42,
+        False,
+    ],
 ]
-        # {
-        #     "image": "assets/image/100.png",
-        # },
 
 
 os.makedirs(TMP_DIR, exist_ok=True)
@@ -126,11 +153,6 @@ def run_refinement(
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    # if not isinstance(rgb_image, Image.Image) and "image" in rgb_image:
-    #     rgb_image = rgb_image["image"]
-
-    # print("rgb_image",rgb_image)
-    
     scene = run_detailgen3d(
         pipeline,
         rgb_image,
@@ -180,7 +202,7 @@ with gr.Blocks() as demo:
                     step=0.1,
                     value=4.0,
                 )
-            gen_button = gr.Button("Generate details", variant="primary")
+            gen_button = gr.Button("Generate Details", variant="primary")
        
         with gr.Column():
             mesh = gr.Model3D(label="Input Coarse Model",camera_position=(90,90,3))