xinjie.wang commited on
Commit
95a94b9
·
1 Parent(s): 79b7dd9
Files changed (2) hide show
  1. app.py +1 -1
  2. common.py +3 -0
app.py CHANGED
@@ -41,7 +41,7 @@ with gr.Blocks(
41
  format="png",
42
  image_mode="RGBA",
43
  type="pil",
44
- height=300,
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()