diff --git a/zoedepth/models/base_models/midas.py b/zoedepth/models/base_models/midas.py
index e26f85895..ec9478c73 100644
--- a/zoedepth/models/base_models/midas.py
+++ b/zoedepth/models/base_models/midas.py
@@ -170,6 +170,8 @@ def get_size(self, width, height):
 
     def __call__(self, x):
         width, height = self.get_size(*x.shape[-2:][::-1])
+        height = int(height)
+        width = int(width)
         return nn.functional.interpolate(x, (height, width), mode='bilinear', align_corners=True)
 
 class PrepForMidas(object):