Update app.py

app.py

@@ -95,8 +95,8 @@ def infer(rgb_image: Image.Image, lr_depth: Image.Image, model_type: str):
         out, _ = net(x_query=lr, rgb=image)
 
     pred = out[0, 0] * (max_out - min_out) + min_out
-    # pred = pred.cpu().numpy().astype(np.uint16)
-    pred = pred.cpu().numpy().astype(np.uint8) 
+    pred = pred.cpu().numpy().astype(np.uint16)
+    
     # raw
     pred_gray = Image.fromarray(pred)