Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 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])
|