abreza commited on
Commit
6e2d18a
1 Parent(s): a3c7a50

fix example images

Browse files
Files changed (1) hide show
  1. launch/image_generation.py +3 -2
launch/image_generation.py CHANGED
@@ -6,7 +6,8 @@ import spaces
6
  import torch
7
  from diffusers import DiffusionPipeline
8
 
9
- from instantMesh.src.utils.infer_util import (remove_background, resize_foreground)
 
10
 
11
 
12
  pipe = DiffusionPipeline.from_pretrained(
@@ -97,7 +98,7 @@ def image_generation_ui():
97
  "Remove Background", elem_id="process", variant="primary")
98
  with gr.Row(variant="panel"):
99
  gr.Examples(
100
- examples=[os.path.join("examples", img_name)
101
  for img_name in sorted(os.listdir("examples/images"))],
102
  inputs=[input_image],
103
  label="Examples",
 
6
  import torch
7
  from diffusers import DiffusionPipeline
8
 
9
+ from instantMesh.src.utils.infer_util import (
10
+ remove_background, resize_foreground)
11
 
12
 
13
  pipe = DiffusionPipeline.from_pretrained(
 
98
  "Remove Background", elem_id="process", variant="primary")
99
  with gr.Row(variant="panel"):
100
  gr.Examples(
101
+ examples=[os.path.join("examples/images", img_name)
102
  for img_name in sorted(os.listdir("examples/images"))],
103
  inputs=[input_image],
104
  label="Examples",