Update app.py

app.py

@@ -38,26 +38,19 @@ model_configs = {
     'vitg': {'encoder': 'vitg', 'features': 384, 'out_channels': [1536, 1536, 1536, 1536]}
 }
 
-ratios_map =  {
-    0.5:{"width":704,"height":1408},
-    0.57:{"width":768,"height":1344},
-    0.68:{"width":832,"height":1216},
-    0.72:{"width":832,"height":1152},
-    0.78:{"width":896,"height":1152},
-    0.82:{"width":896,"height":1088},
-    0.88:{"width":960,"height":1088},
-    0.94:{"width":960,"height":1024},
-    1.00:{"width":1024,"height":1024},
-    1.13:{"width":1088,"height":960},
-    1.21:{"width":1088,"height":896},
-    1.29:{"width":1152,"height":896},
-    1.38:{"width":1152,"height":832},
-    1.46:{"width":1216,"height":832},
-    1.67:{"width":1280,"height":768},
-    1.75:{"width":1344,"height":768},
-    2.00:{"width":1408,"height":704}
+RATIO_CONFIGS_1024 = {
+    0.6666666666666666: {"width": 832, "height": 1248},
+    0.7432432432432432: {"width": 880, "height": 1184},
+    0.8028169014084507: {"width": 912, "height": 1136},
+    1.0: {"width": 1024, "height": 1024},
+    1.2456140350877194: {"width": 1136, "height": 912},
+    1.3454545454545455: {"width": 1184, "height": 880},
+    1.4339622641509433: {"width": 1216, "height": 848},
+    1.5: {"width": 1248, "height": 832},
+    1.5490196078431373: {"width": 1264, "height": 816},
+    1.62: {"width": 1296, "height": 800},
+    1.7708333333333333: {"width": 1360, "height": 768},
 }
-ratios = np.array(list(ratios_map.keys()))
 
 encoder = 'vitl'
 model = DepthAnythingV2(**model_configs[encoder])
@@ -154,19 +147,12 @@ def tile(downscale_factor, input_image):
     control_image = input_image.resize((input_image.size[0] // downscale_factor, input_image.size[1] // downscale_factor)).resize(input_image.size, Image.NEAREST)
     return control_image
     
-def get_size(init_image):
-    w,h=init_image.size
-    curr_ratio = w/h
-    ind = np.argmin(np.abs(curr_ratio-ratios))
-    ratio = ratios[ind]
-    chosen_ratio  = ratios_map[ratio]
-    w,h = chosen_ratio['width'], chosen_ratio['height']
-    return w,h
-
-def resize_img(image):
-    image = image.convert('RGB')
-    w,h = get_size(image)
-    resized_image = image.resize((w, h))
+def resize_img(control_image):
+    image_ratio = control_image.width / control_image.height
+    ratio = min(RATIO_CONFIGS_1024.keys(), key=lambda k: abs(k - image_ratio))
+    to_height = RATIO_CONFIGS_1024[ratio]["height"]
+    to_width = RATIO_CONFIGS_1024[ratio]["width"]
+    resized_image = control_image.resize((to_width, to_height), resample=Image.Resampling.LANCZOS)
     return resized_image
 
 @spaces.GPU(duration=180)