Update app.py

app.py

@@ -7,7 +7,7 @@ import spaces
 import gradio as gr
 import numpy as np
 import torch
-# from gradio_image_prompter import ImagePrompter
+from gradio_image_prompter import ImagePrompter
 from gradio_litmodel3d import LitModel3D
 from huggingface_hub import snapshot_download
 from PIL import Image
@@ -31,7 +31,12 @@ MARKDOWN = """
 3. If you want the refine result to be more consistent with the image, please manually increase the CFG strength.
 """
 EXAMPLES = [
-    ["assets/image/100.png","assets/model/100.glb",42,False,]
+        {
+            "image": "assets/image/100.png",
+        },
+        "assets/model/100.glb",
+        42,
+        False,
 ]
 # EXAMPLES = [
 #     [
@@ -125,7 +130,7 @@ def run_detailgen3d(
     guidance_scale,
 ):
     surface = load_mesh(mesh)
-    image = Image.open(image).convert("RGB")
+    # image = Image.open(image).convert("RGB")
 
     batch_size = 1
 
@@ -168,8 +173,8 @@ def run_refinement(
     # print("rgb_image", rgb_image)
     # print("mesh", rgb_image)
 
-    # if not isinstance(rgb_image, Image.Image) and "image" in rgb_image:
-    #     rgb_image = Image.open(rgb_image["image"]).convert("RGB")
+    if not isinstance(rgb_image, Image.Image) and "image" in rgb_image:
+        rgb_image = Image.open(rgb_image["image"]).convert("RGB")
 
     scene = run_detailgen3d(
         pipeline,
@@ -194,8 +199,8 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             with gr.Row():
-                # image_prompts = ImagePrompter(label="Input Image", type="pil")
-                image_prompts = gr.Image(label="Example Image", type="pil")
+                image_prompts = ImagePrompter(label="Input Image", type="pil")
+                # image_prompts = gr.Image(label="Example Image", type="pil")
                 mesh = gr.Model3D(label="Input Coarse Model",camera_position=(90,90,3))
 
             with gr.Accordion("Generation Settings", open=False):