update real-esrgan to upscale images using tile size of 1000 px

handler.py

@@ -21,6 +21,7 @@ class EndpointHandler:
         self.model = RealESRGANer(
             scale=4,
             model_path=f"/repository/weights/Real-ESRGAN-x4plus.pth",
+            # model_path=f"/workspace/real-esrgan/weights/Real-ESRGAN-x4plus.pth",
             # dni_weight=dni_weight,
             model= RRDBNet(num_in_ch=3, 
                            num_out_ch=3, 
@@ -29,8 +30,8 @@ class EndpointHandler:
                            num_grow_ch=32, 
                            scale=4
                            ),
-            tile=0,
-            tile_pad=10,
+            tile=1000,
+            tile_pad=20,
             # pre_pad=args.pre_pad,
             half=True,
             # gpu_id=args.gpu_id
@@ -52,7 +53,6 @@ class EndpointHandler:
             
             # check image size and mode and return dict
             assert in_mode in ["RGB", "RGBA", "L"], f"Unsupported image mode: {in_mode}"
-            assert in_size[0] * in_size[1] <  1400*1400, f"Image is too large: {in_size}: {in_size[0] * in_size[1]} is greater than {1400*1400}"
             assert outscale > 1 and outscale <=10, f"Outscale must be between 1 and 10: {outscale}"
             
             # debug

unit_tests.py

@@ -74,15 +74,6 @@ class TestEndpointHandler(unittest.TestCase):
         self.assertIn("out_image", result)
         self.assertIsNone(result["error"])
 
-    def test_image_too_large(self):
-        test_image = self.create_test_image(size=(1500, 1500))
-        input_data = {"inputs": {"image": test_image}}
-        
-        result = self.handler(input_data)
-        
-        self.assertIsNone(result["out_image"])
-        self.assertIn("Image is too large", result["error"])
-
     def test_missing_image_key(self):
         input_data = {"inputs": {}}