Update app.py

app.py

@@ -80,20 +80,13 @@ with gr.Blocks(css=css) as demo:
     cmap = matplotlib.colormaps.get_cmap('Spectral_r')
 
     def on_submit(image):
-        if isinstance(image, np.ndarray):
-            image = Image.fromarray(image)
-    
         original_image = image.copy()
-    
-        h, w = image.size  # For PIL images, use .size instead of .shape
-    
-        depth = predict_depth(image)
-    
-        # Debugging info
-        print("Type of depth:", type(depth))
-        print("Shape of depth:", depth.shape if isinstance(depth, np.ndarray) else "N/A")
-    
-        raw_depth = depth
+
+        h, w = image.shape[:2]
+
+        depth = predict_depth(image[:, :, ::-1])
+
+        raw_depth = Image.fromarray(depth.astype('uint16'))
         tmp_raw_depth = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
         raw_depth.save(tmp_raw_depth.name)