pablo commited on
Commit
b785abb
·
1 Parent(s): 38e04f9
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -211,11 +211,11 @@ def predict_images_3d(dict, depth, prompt="", negative_prompt="", guidance_scale
211
  output_image = output.rgb[0]
212
 
213
  mesh_depth = output_depth_vis.max() - output_depth_vis
214
- output_mesh = get_mesh(mesh_depth, output_image, keep_edges=keep_edges)
215
 
216
  depth_image = np.array(depth_image)
217
  depth_image_mesh = depth_image.max() - depth_image
218
- input_mesh = get_mesh(depth_image_mesh,init_image, keep_edges=keep_edges)
219
 
220
  return input_mesh, output_mesh, gr.update(visible=True)
221
 
 
211
  output_image = output.rgb[0]
212
 
213
  mesh_depth = output_depth_vis.max() - output_depth_vis
214
+ output_mesh = get_mesh(mesh_depth, output_image, keep_edges=keep_edges, skew=4)
215
 
216
  depth_image = np.array(depth_image)
217
  depth_image_mesh = depth_image.max() - depth_image
218
+ input_mesh = get_mesh(depth_image_mesh,init_image, keep_edges=keep_edges, skew=4)
219
 
220
  return input_mesh, output_mesh, gr.update(visible=True)
221