anvilinteractiv commited on
Commit
9548c9b
·
verified ·
1 Parent(s): bd3d554

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -204,7 +204,7 @@ def run_full(image: str, seed: int = 0, num_inference_steps: int = 50, guidance_
204
  ).images
205
 
206
  torch.cuda.empty_cache()
207
-
208
  mv_image_path = os.path.join(save_dir, f"mv_adapter_{get_random_hex()}.png")
209
  make_image_grid(images, rows=1).save(mv_image_path)
210
 
@@ -457,6 +457,7 @@ def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
457
 
458
  torch.cuda.empty_cache()
459
  save_dir = os.path.join(TMP_DIR, str(req.session_hash))
 
460
  mv_image_path = os.path.join(save_dir, f"mv_adapter_{get_random_hex()}.png")
461
  make_image_grid(images, rows=1).save(mv_image_path)
462
 
 
204
  ).images
205
 
206
  torch.cuda.empty_cache()
207
+ os.makedirs(save_dir, exist_ok=True)
208
  mv_image_path = os.path.join(save_dir, f"mv_adapter_{get_random_hex()}.png")
209
  make_image_grid(images, rows=1).save(mv_image_path)
210
 
 
457
 
458
  torch.cuda.empty_cache()
459
  save_dir = os.path.join(TMP_DIR, str(req.session_hash))
460
+ os.makedirs(save_dir, exist_ok=True)
461
  mv_image_path = os.path.join(save_dir, f"mv_adapter_{get_random_hex()}.png")
462
  make_image_grid(images, rows=1).save(mv_image_path)
463