Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -255,10 +255,7 @@ def generate_results_and_mesh(image, session_id=None):
|
|
| 255 |
status4, track_img, _ = step4_track(session_id)
|
| 256 |
# Locate mesh (.ply)
|
| 257 |
mesh_dir = os.path.join(os.environ["PIXEL3DMM_TRACKING_OUTPUT"], session_id, "mesh")
|
| 258 |
-
|
| 259 |
-
mesh_file = os.path.relpath(mesh_file_abs, os.getcwd())
|
| 260 |
-
|
| 261 |
-
print(f"[DEBUG] returning mesh_file = {mesh_file}")
|
| 262 |
|
| 263 |
final_status = "\n".join([status1, status2, status3, status4])
|
| 264 |
return final_status, crop_img, normals_img, uv_img, track_img, mesh_file
|
|
@@ -323,7 +320,8 @@ with gr.Blocks(css=css) as demo:
|
|
| 323 |
uv_img = gr.Image(label="UV Map", height=256)
|
| 324 |
track_img = gr.Image(label="Tracking", height=256)
|
| 325 |
with gr.Tab("3D Model"):
|
| 326 |
-
|
|
|
|
| 327 |
|
| 328 |
examples = gr.Examples(
|
| 329 |
examples=[
|
|
|
|
| 255 |
status4, track_img, _ = step4_track(session_id)
|
| 256 |
# Locate mesh (.ply)
|
| 257 |
mesh_dir = os.path.join(os.environ["PIXEL3DMM_TRACKING_OUTPUT"], session_id, "mesh")
|
| 258 |
+
mesh_file = first_file_from_dir(mesh_dir, "glb")
|
|
|
|
|
|
|
|
|
|
| 259 |
|
| 260 |
final_status = "\n".join([status1, status2, status3, status4])
|
| 261 |
return final_status, crop_img, normals_img, uv_img, track_img, mesh_file
|
|
|
|
| 320 |
uv_img = gr.Image(label="UV Map", height=256)
|
| 321 |
track_img = gr.Image(label="Tracking", height=256)
|
| 322 |
with gr.Tab("3D Model"):
|
| 323 |
+
with gr.Column():
|
| 324 |
+
mesh_file = gr.Model3D(label="3D Model Preview", height=512)
|
| 325 |
|
| 326 |
examples = gr.Examples(
|
| 327 |
examples=[
|