ReubenSun commited on
Commit
0605002
·
1 Parent(s): 56b8892

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, args.device
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