pablo commited on
Commit
fd28d2d
·
1 Parent(s): fd59cc1
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -215,10 +215,10 @@ def predict_images_3d(dict, depth, prompt="", negative_prompt="", guidance_scale
215
  output_image = output.rgb[0]
216
 
217
  mesh_depth = output_depth_vis.max() - output_depth_vis
218
- output_mesh = get_mesh(mesh_depth, output_image, keep_edges=keep_edges, skew=4)
219
 
220
  depth_image_mesh = input_depth_vis.max() - input_depth_vis
221
- input_mesh = get_mesh(depth_image_mesh,init_image, keep_edges=keep_edges, skew=4)
222
 
223
  return input_mesh, output_mesh, gr.update(visible=True)
224
 
 
215
  output_image = output.rgb[0]
216
 
217
  mesh_depth = output_depth_vis.max() - output_depth_vis
218
+ output_mesh = get_mesh(mesh_depth, output_image, keep_edges=keep_edges, skew=1)
219
 
220
  depth_image_mesh = input_depth_vis.max() - input_depth_vis
221
+ input_mesh = get_mesh(depth_image_mesh,init_image, keep_edges=keep_edges, skew=1)
222
 
223
  return input_mesh, output_mesh, gr.update(visible=True)
224