alexnasa commited on
Commit
e0832eb
·
verified ·
1 Parent(s): 4c2fa10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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
- mesh_file_abs = first_file_from_dir(mesh_dir, "ply")
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
- mesh_file = gr.Model3D(label="3D Model Preview")
 
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=[