Update app.py

app.py

@@ -66,11 +66,14 @@ def predict_depth(input_image):
         # Clip inverse depth to 0-10 range
         inverse_depth_clipped = np.clip(inverse_depth, 0, 10)
 
+        # Normalize
+        norm_depth = np.log(depth)
+
         # Create a color map
         plt.figure(figsize=(15.36, 15.36), dpi=100)  # Set figure size to 1536x1536 pixels
-        plt.imshow(inverse_depth_clipped, cmap='viridis')
-        plt.colorbar(label='Inverse Depth')
-        plt.title('Predicted Inverse Depth Map')
+        plt.imshow(norm_depth, cmap='viridis')
+        plt.colorbar(label='Normalized Depth')
+        plt.title('Predicted Normalized Depth Map')
         plt.axis('off')
         
         # Save the plot to a file