Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
95a94b9
1
Parent(s):
79b7dd9
update
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ with gr.Blocks(
|
|
41 |
format="png",
|
42 |
image_mode="RGBA",
|
43 |
type="pil",
|
44 |
-
height=
|
45 |
)
|
46 |
gr.Markdown(
|
47 |
"""
|
|
|
41 |
format="png",
|
42 |
image_mode="RGBA",
|
43 |
type="pil",
|
44 |
+
height=500,
|
45 |
)
|
46 |
gr.Markdown(
|
47 |
"""
|
common.py
CHANGED
@@ -359,6 +359,9 @@ def image_to_3d(
|
|
359 |
if isinstance(seg_image, np.ndarray):
|
360 |
seg_image = Image.fromarray(seg_image)
|
361 |
|
|
|
|
|
|
|
362 |
seg_image.save(f"{TMP_DIR}/{req.session_hash}/seg_image.png")
|
363 |
raw_image_cache.save(f"{TMP_DIR}/{req.session_hash}/raw_image.png")
|
364 |
PIPELINE.cuda()
|
|
|
359 |
if isinstance(seg_image, np.ndarray):
|
360 |
seg_image = Image.fromarray(seg_image)
|
361 |
|
362 |
+
print("raw_image_cache", raw_image_cache)
|
363 |
+
print("seg_image", seg_image)
|
364 |
+
os.makedirs(f"{TMP_DIR}/{req.session_hash}", exist_ok=True)
|
365 |
seg_image.save(f"{TMP_DIR}/{req.session_hash}/seg_image.png")
|
366 |
raw_image_cache.save(f"{TMP_DIR}/{req.session_hash}/raw_image.png")
|
367 |
PIPELINE.cuda()
|