mehdizz commited on
Commit
9e55e75
·
verified ·
1 Parent(s): b3c56d4

os.makedirs(..............., exist_ok=True)

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -28,6 +28,7 @@ def start_session(req: gr.Request):
28
 
29
  def end_session(req: gr.Request):
30
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
 
31
  shutil.rmtree(user_dir)
32
 
33
 
@@ -136,6 +137,7 @@ def image_to_3d(
136
  str: The path to the video of the 3D model.
137
  """
138
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
 
139
  outputs = pipeline.run(
140
  image,
141
  seed=seed,
@@ -154,6 +156,7 @@ def image_to_3d(
154
  video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
155
  video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
156
  video_path = os.path.join(user_dir, 'sample.mp4')
 
157
  imageio.mimsave(video_path, video, fps=15)
158
  state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
159
  torch.cuda.empty_cache()
@@ -179,9 +182,11 @@ def extract_glb(
179
  str: The path to the extracted GLB file.
180
  """
181
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
 
182
  gs, mesh = unpack_state(state)
183
  glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
184
  glb_path = os.path.join(user_dir, 'sample.glb')
 
185
  glb.export(glb_path)
186
  torch.cuda.empty_cache()
187
  return glb_path, glb_path
 
28
 
29
  def end_session(req: gr.Request):
30
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
31
+ os.makedirs(user_dir, exist_ok=True)
32
  shutil.rmtree(user_dir)
33
 
34
 
 
137
  str: The path to the video of the 3D model.
138
  """
139
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
140
+ os.makedirs(user_dir, exist_ok=True)
141
  outputs = pipeline.run(
142
  image,
143
  seed=seed,
 
156
  video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
157
  video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
158
  video_path = os.path.join(user_dir, 'sample.mp4')
159
+ os.makedirs(video_path, exist_ok=True)
160
  imageio.mimsave(video_path, video, fps=15)
161
  state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
162
  torch.cuda.empty_cache()
 
182
  str: The path to the extracted GLB file.
183
  """
184
  user_dir = os.path.join(TMP_DIR, str(req.session_hash))
185
+ os.makedirs(user_dir, exist_ok=True)
186
  gs, mesh = unpack_state(state)
187
  glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
188
  glb_path = os.path.join(user_dir, 'sample.glb')
189
+ os.makedirs(glb_path, exist_ok=True)
190
  glb.export(glb_path)
191
  torch.cuda.empty_cache()
192
  return glb_path, glb_path