Update app.py

app.py

@@ -105,7 +105,8 @@ def infer(rgb_image: Image.Image, lr_depth: Image.Image, model_type: str):
     pred_vis = pred_norm.astype(np.uint8)
     pred_heat = cv2.applyColorMap(pred_vis, cv2.COLORMAP_PLASMA)
     pred_heat = cv2.cvtColor(pred_heat, cv2.COLOR_BGR2RGB)
-    return pred_gray, Image.fromarray(pred_heat)
+    # return pred_gray, Image.fromarray(pred_heat)
+    return Image.fromarray(pred_heat)
 
 
 Intro = """