redundant code, max width/height

app.py

@@ -37,62 +37,6 @@ def eraser_api_call(image_base64_file, mask_base64_file, mask_type):
     
     return res_image
 
-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}
-}
-ratios = np.array(list(ratios_map.keys()))
-
-
-
-def get_masked_image(image, image_mask, width, height):
-    image_mask = image_mask # inpaint area is white
-    image_mask = image_mask.resize((width, height)) # object to remove is white (1)
-    image_mask_pil = image_mask
-    image = np.array(image.convert("RGB")).astype(np.float32) / 255.0
-    image_mask = np.array(image_mask_pil.convert("L")).astype(np.float32) / 255.0
-    assert image.shape[0:1] == image_mask.shape[0:1], "image and image_mask must have the same image size"
-    masked_image_to_present = image.copy()
-    masked_image_to_present[image_mask > 0.5] = (0.5,0.5,0.5)  # set as masked pixel
-    image[image_mask > 0.5] = 0.5  # set as masked pixel - s.t. will be grey 
-    image = Image.fromarray((image * 255.0).astype(np.uint8))
-    masked_image_to_present = Image.fromarray((masked_image_to_present * 255.0).astype(np.uint8))
-    return image, image_mask_pil, masked_image_to_present
-
-    
-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 read_content(file_path: str) -> str:
-    """read the content of target file
-    """
-    with open(file_path, 'r', encoding='utf-8') as f:
-        content = f.read()
-
-    return content
 
 def predict(dict):
 
@@ -162,13 +106,16 @@ with image_blocks as demo:
         ''')
     with gr.Row():
         with gr.Column():
-            image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], brush=gr.Brush(colors=["#000000"], color_mode="fixed"))
+            image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
+                                   brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
+                                   max_width=512, max_height=512
+                                   )
             with gr.Row(elem_id="prompt-container", equal_height=True):
                 with gr.Column():  # Wrap the button inside a Column
                     btn = gr.Button("Erase!", elem_id="run_button")
         
         with gr.Column():
-            image_out = gr.Image(label="Output", elem_id="output-img", height=400)
+            image_out = gr.Image(label="Output", elem_id="output-img", max_width=512, max_height=512)
 
     # Button click will trigger the inpainting function (no prompt required)
     btn.click(fn=predict, inputs=[image], outputs=[image_out], api_name='run')