Spaces:
Running
on
Zero
Running
on
Zero
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 |
|