Spaces:
Running
on
Zero
Running
on
Zero
logging for textrure fail
Browse files
app.py
CHANGED
@@ -151,12 +151,26 @@ def extract_glb(
|
|
151 |
"""
|
152 |
user_dir = os.path.join(TMP_DIR, str(req.session_hash))
|
153 |
os.makedirs(user_dir, exist_ok=True)
|
154 |
-
|
155 |
-
|
156 |
-
|
157 |
-
|
158 |
-
|
159 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
160 |
|
161 |
|
162 |
@spaces.GPU
|
|
|
151 |
"""
|
152 |
user_dir = os.path.join(TMP_DIR, str(req.session_hash))
|
153 |
os.makedirs(user_dir, exist_ok=True)
|
154 |
+
logging.info(f"[{str(req.session_hash)[:8]}] ENTER extract_glb - Simplify: {mesh_simplify}, Texture Size: {texture_size}")
|
155 |
+
try:
|
156 |
+
gs, mesh = unpack_state(state)
|
157 |
+
logging.info(f"[{str(req.session_hash)[:8]}] State unpacked successfully.")
|
158 |
+
|
159 |
+
# Call to_glb with verbose=True
|
160 |
+
logging.info(f"[{str(req.session_hash)[:8]}] Calling postprocessing_utils.to_glb with verbose=True...")
|
161 |
+
glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=True)
|
162 |
+
logging.info(f"[{str(req.session_hash)[:8]}] postprocessing_utils.to_glb completed.")
|
163 |
+
|
164 |
+
glb_path = os.path.join(user_dir, 'sample.glb')
|
165 |
+
glb.export(glb_path)
|
166 |
+
logging.info(f"[{str(req.session_hash)[:8]}] GLB exported to {glb_path}")
|
167 |
+
torch.cuda.empty_cache()
|
168 |
+
logging.info(f"[{str(req.session_hash)[:8]}] EXIT extract_glb - Success")
|
169 |
+
return glb_path, glb_path
|
170 |
+
except Exception as e:
|
171 |
+
logging.error(f"[{str(req.session_hash)[:8]}] ERROR in extract_glb: {e}", exc_info=True)
|
172 |
+
torch.cuda.empty_cache()
|
173 |
+
raise gr.Error(f"GLB Extraction failed: {e}")
|
174 |
|
175 |
|
176 |
@spaces.GPU
|