Update app.py

app.py

@@ -286,7 +286,7 @@ def generate_images_30(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
     # 4. Create 16-bit PIL Image from the same tensor (for Upscaler)
     # Pillow 10+ can create an 'RGB' mode image from a uint16 array
     srgb_numpy_16bit = (srgb_tensor_0_1.cpu().permute(0, 2, 3, 1).float().numpy()[0] * 65535.0).round().astype("uint16")
-    sd_image_pil_16bit = Image.fromarray(srgb_numpy_16bit, mode='RGB')
+    sd_image_pil_16bit = Image.fromarray(srgb_numpy_16bit)
     print('-- got 16-bit PIL image for upscaling --')
 
     # 5. Run the 16-bit upscaling (4x)
@@ -295,8 +295,6 @@ def generate_images_30(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
         upscale_1 = upscaler_2(sd_image_pil_16bit, tiling=True, tile_width=256, tile_height=256)
         upscale_2 = upscaler_2(upscale_1, tiling=True, tile_width=256, tile_height=256)
     print('-- got 4K 16-bit upscaled PIL image --')
-    
-    #downscaled_upscale = upscale_2.resize((upscale_2.width // 8, upscale_2.height // 8), Image.LANCZOS)
 
     torch.cuda.empty_cache()
     
@@ -357,8 +355,6 @@ def generate_images_60(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
         upscale_1 = upscaler_2(sd_image_pil_16bit, tiling=True, tile_width=256, tile_height=256)
         upscale_2 = upscaler_2(upscale_1, tiling=True, tile_width=256, tile_height=256)
     print('-- got 4K 16-bit upscaled PIL image --')
-    
-    #downscaled_upscale = upscale_2.resize((upscale_2.width // 8, upscale_2.height // 8), Image.LANCZOS)
 
     torch.cuda.empty_cache()
     
@@ -413,14 +409,12 @@ def generate_images_110(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
     sd_image_pil_16bit = Image.fromarray(srgb_numpy_16bit, mode='RGB')
     print('-- got 16-bit PIL image for upscaling --')
 
-   # 5. Run the 16-bit upscaling (4x)
+    # 5. Run the 16-bit upscaling (4x)
     # We feed the high-precision 16-bit PIL image to the upscaler
     with torch.no_grad():
         upscale_1 = upscaler_2(sd_image_pil_16bit, tiling=True, tile_width=256, tile_height=256)
         upscale_2 = upscaler_2(upscale_1, tiling=True, tile_width=256, tile_height=256)
     print('-- got 4K 16-bit upscaled PIL image --')
-    
-    #downscaled_upscale = upscale_2.resize((upscale_2.width // 8, upscale_2.height // 8), Image.LANCZOS)
 
     torch.cuda.empty_cache()