hysts HF Staff commited on
Commit
76fd876
·
1 Parent(s): 7c8a18f
Files changed (1) hide show
  1. app.py +7 -9
app.py CHANGED
@@ -57,8 +57,7 @@ def preprocess_image(image: Image.Image) -> Image.Image:
57
  Returns:
58
  Image.Image: The preprocessed image.
59
  """
60
- processed_image = pipeline.preprocess_image(image)
61
- return processed_image
62
 
63
 
64
  def preprocess_images(images: list[tuple[Image.Image, str]]) -> list[Image.Image]:
@@ -71,8 +70,7 @@ def preprocess_images(images: list[tuple[Image.Image, str]]) -> list[Image.Image
71
  List[Image.Image]: The preprocessed images.
72
  """
73
  images = [image[0] for image in images]
74
- processed_images = [pipeline.preprocess_image(image) for image in images]
75
- return processed_images
76
 
77
 
78
  def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
@@ -335,7 +333,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
335
  download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
336
  download_gs = gr.DownloadButton(label="Download Gaussian", interactive=False)
337
 
338
- is_multiimage = gr.State(False)
339
  output_buf = gr.State()
340
 
341
  # Example images at the bottom of the page
@@ -363,11 +361,11 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
363
  demo.unload(end_session)
364
 
365
  single_image_input_tab.select(
366
- lambda: tuple([False, gr.Row.update(visible=True), gr.Row.update(visible=False)]),
367
  outputs=[is_multiimage, single_image_example, multiimage_example],
368
  )
369
  multiimage_input_tab.select(
370
- lambda: tuple([True, gr.Row.update(visible=False), gr.Row.update(visible=True)]),
371
  outputs=[is_multiimage, single_image_example, multiimage_example],
372
  )
373
 
@@ -401,12 +399,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
401
  ],
402
  outputs=[output_buf, video_output],
403
  ).then(
404
- lambda: tuple([gr.Button(interactive=True), gr.Button(interactive=True)]),
405
  outputs=[extract_glb_btn, extract_gs_btn],
406
  )
407
 
408
  video_output.clear(
409
- lambda: tuple([gr.Button(interactive=False), gr.Button(interactive=False)]),
410
  outputs=[extract_glb_btn, extract_gs_btn],
411
  )
412
 
 
57
  Returns:
58
  Image.Image: The preprocessed image.
59
  """
60
+ return pipeline.preprocess_image(image)
 
61
 
62
 
63
  def preprocess_images(images: list[tuple[Image.Image, str]]) -> list[Image.Image]:
 
70
  List[Image.Image]: The preprocessed images.
71
  """
72
  images = [image[0] for image in images]
73
+ return [pipeline.preprocess_image(image) for image in images]
 
74
 
75
 
76
  def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
 
333
  download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
334
  download_gs = gr.DownloadButton(label="Download Gaussian", interactive=False)
335
 
336
+ is_multiimage = gr.State(False) # noqa: FBT003
337
  output_buf = gr.State()
338
 
339
  # Example images at the bottom of the page
 
361
  demo.unload(end_session)
362
 
363
  single_image_input_tab.select(
364
+ lambda: (False, gr.Row.update(visible=True), gr.Row.update(visible=False)),
365
  outputs=[is_multiimage, single_image_example, multiimage_example],
366
  )
367
  multiimage_input_tab.select(
368
+ lambda: (True, gr.Row.update(visible=False), gr.Row.update(visible=True)),
369
  outputs=[is_multiimage, single_image_example, multiimage_example],
370
  )
371
 
 
399
  ],
400
  outputs=[output_buf, video_output],
401
  ).then(
402
+ lambda: (gr.Button(interactive=True), gr.Button(interactive=True)),
403
  outputs=[extract_glb_btn, extract_gs_btn],
404
  )
405
 
406
  video_output.clear(
407
+ lambda: (gr.Button(interactive=False), gr.Button(interactive=False)),
408
  outputs=[extract_glb_btn, extract_gs_btn],
409
  )
410