Spaces:
Runtime error
Runtime error
lionelgarnier
commited on
Commit
·
d5ea645
1
Parent(s):
5c85703
update user directory handling to use a fixed temp_output path
Browse files
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,
|
|
|
59 |
os.makedirs(user_dir, exist_ok=True)
|
60 |
|
61 |
def end_session(req: gr.Request):
|
62 |
-
user_dir = os.path.join(TMP_DIR,
|
|
|
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))]
|