lionelgarnier commited on
Commit
d5ea645
·
1 Parent(s): 5c85703

update user directory handling to use a fixed temp_output path

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -55,11 +55,13 @@ _trellis_pipeline = None
55
 
56
 
57
  def start_session(req: gr.Request):
58
- user_dir = os.path.join(TMP_DIR, str(req.session_hash))
 
59
  os.makedirs(user_dir, exist_ok=True)
60
 
61
  def end_session(req: gr.Request):
62
- user_dir = os.path.join(TMP_DIR, str(req.session_hash))
 
63
  shutil.rmtree(user_dir)
64
 
65
  def preprocess_image(image: Image.Image) -> Image.Image:
@@ -345,7 +347,8 @@ def image_to_3d(
345
  "cfg_strength": slat_guidance_strength,
346
  },
347
  )
348
-
 
349
  video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
350
  video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
351
  video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
 
55
 
56
 
57
  def start_session(req: gr.Request):
58
+ user_dir = os.path.join(TMP_DIR, "temp_output")
59
+ # user_dir = os.path.join(TMP_DIR, str(req.session_hash))
60
  os.makedirs(user_dir, exist_ok=True)
61
 
62
  def end_session(req: gr.Request):
63
+ user_dir = os.path.join(TMP_DIR, "temp_output")
64
+ # user_dir = os.path.join(TMP_DIR, str(req.session_hash))
65
  shutil.rmtree(user_dir)
66
 
67
  def preprocess_image(image: Image.Image) -> Image.Image:
 
347
  "cfg_strength": slat_guidance_strength,
348
  },
349
  )
350
+ temp_dir = os.path.join(TMP_DIR, "temp_output")
351
+
352
  video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
353
  video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
354
  video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]