prettyflacko03 commited on
Commit
2a2c1e0
·
verified ·
1 Parent(s): 8bd98db

Update app.py

Browse files

remove is_multiimage

Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -128,7 +128,6 @@ def get_seed(randomize_seed: bool, seed: int) -> int:
128
  def generate_and_extract_glb(
129
  image: Image.Image,
130
  multiimages: List[Tuple[Image.Image, str]],
131
- is_multiimage: bool,
132
  seed: int,
133
  ss_guidance_strength: float,
134
  ss_sampling_steps: int,
@@ -164,7 +163,7 @@ def generate_and_extract_glb(
164
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
165
 
166
  # Generate 3D model
167
- if not is_multiimage:
168
  outputs = pipeline.run(
169
  image,
170
  seed=seed,
@@ -390,7 +389,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
390
  outputs=[seed],
391
  ).then(
392
  generate_and_extract_glb,
393
- inputs=[image_prompt, multiimage_prompt, is_multiimage, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps, multiimage_algo, mesh_simplify, texture_size],
394
  outputs=[output_buf, video_output, model_output, download_glb],
395
  ).then(
396
  lambda: tuple([gr.Button(interactive=True), gr.Button(interactive=True)]),
 
128
  def generate_and_extract_glb(
129
  image: Image.Image,
130
  multiimages: List[Tuple[Image.Image, str]],
 
131
  seed: int,
132
  ss_guidance_strength: float,
133
  ss_sampling_steps: int,
 
163
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
164
 
165
  # Generate 3D model
166
+ if image:
167
  outputs = pipeline.run(
168
  image,
169
  seed=seed,
 
389
  outputs=[seed],
390
  ).then(
391
  generate_and_extract_glb,
392
+ inputs=[image_prompt, multiimage_prompt, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps, multiimage_algo, mesh_simplify, texture_size],
393
  outputs=[output_buf, video_output, model_output, download_glb],
394
  ).then(
395
  lambda: tuple([gr.Button(interactive=True), gr.Button(interactive=True)]),