fix:gradio
Browse files
app.py
CHANGED
@@ -100,9 +100,7 @@ with gr.Blocks(title="Step1X-3D demo") as demo:
|
|
100 |
gr.Markdown("# Step1X-3D")
|
101 |
with gr.Row():
|
102 |
with gr.Column(scale=2):
|
103 |
-
input_image = gr.Image(
|
104 |
-
label="Image", type="filepath", image_mode="RGBA"
|
105 |
-
)
|
106 |
guidance_scale = gr.Number(label="Guidance Scale", value="7.5")
|
107 |
inference_steps = gr.Slider(
|
108 |
label="Inferece Steps", minimum=1, maximum=100, value=50
|
|
|
100 |
gr.Markdown("# Step1X-3D")
|
101 |
with gr.Row():
|
102 |
with gr.Column(scale=2):
|
103 |
+
input_image = gr.Image(label="Image", type="filepath")
|
|
|
|
|
104 |
guidance_scale = gr.Number(label="Guidance Scale", value="7.5")
|
105 |
inference_steps = gr.Slider(
|
106 |
label="Inferece Steps", minimum=1, maximum=100, value=50
|
step1x3d_texture/pipelines/step1x_3d_texture_synthesis_pipeline.py
CHANGED
@@ -314,10 +314,6 @@ class Step1X3DTexturePipeline:
|
|
314 |
texture, ori_trust_map = render.fast_bake_texture(
|
315 |
project_textures, project_weighted_cos_maps
|
316 |
)
|
317 |
-
elif method == "poisson":
|
318 |
-
texture = poisson_blend(
|
319 |
-
project_textures, project_weighted_cos_maps, project_boundary_maps
|
320 |
-
)
|
321 |
else:
|
322 |
raise f"no method {method}"
|
323 |
return texture, ori_trust_map > 1e-8
|
@@ -343,7 +339,7 @@ class Step1X3DTexturePipeline:
|
|
343 |
]
|
344 |
)
|
345 |
remove_bg_fn = lambda x: self.remove_bg(
|
346 |
-
x, birefnet, transform_image,
|
347 |
)
|
348 |
else:
|
349 |
remove_bg_fn = None
|
|
|
314 |
texture, ori_trust_map = render.fast_bake_texture(
|
315 |
project_textures, project_weighted_cos_maps
|
316 |
)
|
|
|
|
|
|
|
|
|
317 |
else:
|
318 |
raise f"no method {method}"
|
319 |
return texture, ori_trust_map > 1e-8
|
|
|
339 |
]
|
340 |
)
|
341 |
remove_bg_fn = lambda x: self.remove_bg(
|
342 |
+
x, birefnet, transform_image, self.config.device
|
343 |
)
|
344 |
else:
|
345 |
remove_bg_fn = None
|