anvilinteractiv commited on
Commit
76af743
·
verified ·
1 Parent(s): 54f7867

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -386,7 +386,7 @@ def image_to_3d(
386
  logger.info(f"Saved mesh to {mesh_path}")
387
 
388
  torch.cuda.empty_cache()
389
- return {"path": mesh_path}
390
  except Exception as e:
391
  logger.error(f"Error in image_to_3d: {str(e)}")
392
  raise
@@ -678,7 +678,8 @@ try:
678
  num_inference_steps,
679
  guidance_scale,
680
  reduce_face,
681
- target_face_num
 
682
  ],
683
  outputs=[model_output]
684
  ).then(lambda: gr.Button(interactive=True), outputs=[gen_texture_button])
 
386
  logger.info(f"Saved mesh to {mesh_path}")
387
 
388
  torch.cuda.empty_cache()
389
+ return mesh_path
390
  except Exception as e:
391
  logger.error(f"Error in image_to_3d: {str(e)}")
392
  raise
 
678
  num_inference_steps,
679
  guidance_scale,
680
  reduce_face,
681
+ target_face_num,
682
+ gr.State(value=None) # Pass None for req since web UI doesn't provide it
683
  ],
684
  outputs=[model_output]
685
  ).then(lambda: gr.Button(interactive=True), outputs=[gen_texture_button])