Update app.py

app.py

@@ -549,7 +549,9 @@ class DDIMScheduler():
 
 image_processor, swin_transformer, vae, unet, scheduler = load_models()
 
-def MonoGeoDepthModelRun(image):
+def MonoGeoDepthModelRun(numpy_image):
+    numpy_image = numpy_image.astype(np.uint8)
+    image = Image.fromarray(numpy_image)
     batch_size=1
     torch_device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 
@@ -574,12 +576,14 @@ def MonoGeoDepthModelRun(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))
 
     return predicted_dtm