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