Fix depth image conversion issue

app.py

@@ -110,10 +110,10 @@ with gr.Blocks(css=css) as demo:
 
             # If pred_depth is in an acceptable range for 16-bit representation
             # No need for further normalization
-            pred_depth_16bit = pred_depth_squeezed.cpu().numpy().astype('uint16')
+            # pred_depth_16bit = pred_depth_squeezed.cpu().numpy().astype('uint16')
 
             # Convert to PIL image
-            raw_depth = Image.fromarray(pred_depth_16bit)
+            raw_depth = Image.fromarray(pred_depth_squeezed)
 
             # Continue with your file saving operations
             tmp = tempfile.NamedTemporaryFile(suffix='.png', delete=False)