Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|