Update app.py

app.py

@@ -605,12 +605,7 @@ def infer_upscale(
         # resize to target desired size
         image = image.resize((w_original * upscale_factor, h_original * upscale_factor))
         
-        # Convert the image back to PIL Image and ensure proper scaling
-        if isinstance(image, torch.Tensor):
-            image = image.cpu().permute(1, 2, 0).numpy()
-        if image.max() <= 1.0:
-            image = (image * 255).astype(np.uint8)
-        image = Image.fromarray(image)
+        # image is already a PIL Image, no need for further conversion
 
         return image, seed, num_inference_steps, upscale_factor, controlnet_conditioning_scale, gr.update(), gr.update(visible=False)
     except Exception as e:
@@ -808,6 +803,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
             upscale_randomize_seed,
             upscale_error
         ],
+
     ).then(
         infer_upscale,
         inputs=[