Spaces:
Runtime error
Runtime error
fix example images
Browse files
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 (
|
|
|
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",
|