Update app.py

app.py

@@ -579,15 +579,9 @@ def MonoGeoDepthModelRun(numpy_image):
 
         image_ = predicted_dtm.squeeze(0)
         image_ = (image_ - image_.min()) / (image_.max() - image_.min())
-
-        image_np = image_.numpy()
-        cmap = plt.get_cmap('jet_r')
-        image_colored = cmap(image_np)
-        image_colored = image_colored[:, :, :3]  # Use only the RGB channels
-        predicted_dtm = ToPILImage()(image_colored)
         
-        # to_pil = ToPILImage()
-        # predicted_dtm = to_pil(image_)
+        to_pil = ToPILImage()
+        predicted_dtm = to_pil(image_)
 
     return predicted_dtm