Fix bad size of upscaled image in base 64

ESRGANer.py

@@ -79,6 +79,8 @@ class ESRGANer():
         tiles_x = math.ceil(width / self.tile_size)
         tiles_y = math.ceil(height / self.tile_size)
 
+        print("Image processing started...")
+
         # loop over all tiles
         for y in range(tiles_y):
             for x in range(tiles_x):
@@ -103,8 +105,6 @@ class ESRGANer():
                 tile_idx = y * tiles_x + x + 1
                 input_tile = self.img[:, :, input_start_y_pad:input_end_y_pad, input_start_x_pad:input_end_x_pad]
 
-                print("Image processing started...")
-
                 # upscale tile
                 try:
                     with torch.no_grad():

process_image.py

@@ -4,6 +4,7 @@ from run_cmd import run_cmd
 from PIL import Image
 import tempfile
 import uuid
+import numpy as np
 
 temp_path = tempfile.gettempdir()
 
@@ -25,6 +26,6 @@ def inference(img, size, type):
     if size == "x2":
         img_out = img_out.resize((img_out.width // 2, img_out.height // 2), resample=Image.BICUBIC)
 
-    #img_out.thumbnail((600, 600), Image.ANTIALIAS)
+    img_out = np.array(img_out)
 
     return img_out, gr.File.update(value=OUTPUT_PATH)