linoyts HF staff commited on
Commit
ed3a4d8
·
verified ·
1 Parent(s): 287e155

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -254,7 +254,7 @@ def image_to_3d(
254
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
255
  if not is_multiimage:
256
  outputs = pipeline.run(
257
- image[1]['composite'],
258
  seed=seed,
259
  formats=["gaussian", "mesh"],
260
  preprocess_image=False,
@@ -387,7 +387,6 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
387
  #image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=300)
388
  with gr.Column():
389
  image_prompt = gr.ImageMask(type="pil", image_mode="RGB", height=512, value={"background":Image.new("RGB", (512, 512), (255, 255, 255)), "layers":[Image.new("RGB", (512, 512), (255, 255, 255))], "composite":Image.new("RGB", (512, 512), (255, 255, 255))})
390
- image_prompt_processed = ImageSlider(type="pil", height=512)
391
  with gr.Row():
392
  sketch_btn = gr.Button("process sketch")
393
  generate_btn = gr.Button("Generate 3D")
@@ -456,6 +455,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
456
 
457
  with gr.Column():
458
  video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
 
459
  model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
460
 
461
  with gr.Row():
 
254
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
255
  if not is_multiimage:
256
  outputs = pipeline.run(
257
+ image[1],
258
  seed=seed,
259
  formats=["gaussian", "mesh"],
260
  preprocess_image=False,
 
387
  #image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=300)
388
  with gr.Column():
389
  image_prompt = gr.ImageMask(type="pil", image_mode="RGB", height=512, value={"background":Image.new("RGB", (512, 512), (255, 255, 255)), "layers":[Image.new("RGB", (512, 512), (255, 255, 255))], "composite":Image.new("RGB", (512, 512), (255, 255, 255))})
 
390
  with gr.Row():
391
  sketch_btn = gr.Button("process sketch")
392
  generate_btn = gr.Button("Generate 3D")
 
455
 
456
  with gr.Column():
457
  video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
458
+ image_prompt_processed = ImageSlider(type="pil", height=512)
459
  model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
460
 
461
  with gr.Row():