correct scale

app.py

@@ -206,6 +206,10 @@ def predict_images_3d(dict, depth, prompt="", negative_prompt="", guidance_scale
     output_depth_vis = (depth_out - np.min(depth_out)) / (np.max(depth_out) - np.min(depth_out)) * 255
     output_depth_vis = output_depth_vis.astype("uint8")
 
+    input_depth = np.array(depth_image)
+    input_depth_vis = (input_depth - np.min(input_depth)) / (np.max(input_depth) - np.min(input_depth)) * 255
+    input_depth_vis = input_depth_vis.astype("uint8")
+
     #init_image
     #depth_image
     output_image = output.rgb[0]
@@ -213,8 +217,7 @@ def predict_images_3d(dict, depth, prompt="", negative_prompt="", guidance_scale
     mesh_depth = output_depth_vis.max() - output_depth_vis
     output_mesh = get_mesh(mesh_depth, output_image, keep_edges=keep_edges, skew=4)
 
-    depth_image = np.array(depth_image)
-    depth_image_mesh = depth_image.max() - depth_image
+    depth_image_mesh = input_depth_vis.max() - input_depth_vis
     input_mesh = get_mesh(depth_image_mesh,init_image, keep_edges=keep_edges, skew=4)
     
     return input_mesh, output_mesh, gr.update(visible=True)