Update app.py
Browse files
app.py
CHANGED
@@ -68,9 +68,9 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
|
|
68 |
return image
|
69 |
|
70 |
|
71 |
-
def generate(image
|
72 |
scene_codes = model(image, device=device)
|
73 |
-
mesh = model.extract_mesh(scene_codes
|
74 |
mesh = to_gradio_3d_orientation(mesh)
|
75 |
mesh_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False)
|
76 |
mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
|
@@ -108,13 +108,6 @@ with gr.Blocks() as demo:
|
|
108 |
value=0.85,
|
109 |
step=0.05,
|
110 |
)
|
111 |
-
mc_resolution = gr.Slider(
|
112 |
-
label="Marching Cubes Resolution",
|
113 |
-
minimum=32,
|
114 |
-
maximum=1024,
|
115 |
-
value=256,
|
116 |
-
step=32
|
117 |
-
)
|
118 |
with gr.Row():
|
119 |
submit = gr.Button("Generate", elem_id="generate", variant="primary")
|
120 |
with gr.Column():
|
@@ -146,7 +139,7 @@ with gr.Blocks() as demo:
|
|
146 |
outputs=[processed_image],
|
147 |
).success(
|
148 |
fn=generate,
|
149 |
-
inputs=[processed_image
|
150 |
outputs=[output_model, output_model2],
|
151 |
)
|
152 |
|
|
|
68 |
return image
|
69 |
|
70 |
|
71 |
+
def generate(image):
|
72 |
scene_codes = model(image, device=device)
|
73 |
+
mesh = model.extract_mesh(scene_codes)[0]
|
74 |
mesh = to_gradio_3d_orientation(mesh)
|
75 |
mesh_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False)
|
76 |
mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
|
|
|
108 |
value=0.85,
|
109 |
step=0.05,
|
110 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
111 |
with gr.Row():
|
112 |
submit = gr.Button("Generate", elem_id="generate", variant="primary")
|
113 |
with gr.Column():
|
|
|
139 |
outputs=[processed_image],
|
140 |
).success(
|
141 |
fn=generate,
|
142 |
+
inputs=[processed_image],
|
143 |
outputs=[output_model, output_model2],
|
144 |
)
|
145 |
|