Update app.py

app.py

@@ -576,14 +576,18 @@ def MonoGeoDepthModelRun(numpy_image):
         noisy_latents = model_input - noise_pred[0]
         predicted_dtm = latent_to_tensor(noisy_latents, vae)
         predicted_dtm = predicted_dtm.detach().cpu()
-        print("type 1: ",type(predicted_dtm))
 
         image_ = predicted_dtm.squeeze(0)
         image_ = (image_ - image_.min()) / (image_.max() - image_.min())
 
-        to_pil = ToPILImage()
-        predicted_dtm = to_pil(image_)
-        print("type 2: ",type(predicted_dtm))
+        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_)
 
     return predicted_dtm