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",
|