Update app.py

app.py

@@ -176,9 +176,17 @@ def preprocess_image(input_image: Image.Image) -> Image.Image:
     img = input_image.copy()
     img.thumbnail((256, 256), Image.Resampling.BILINEAR)
     return img
+
+
+@spaces.GPU()
+def preprocess_n_magnify(input_image: Image.Image):
+    preprocessed_img = preprocess_image(input_image)
+    preprocessed_img, magnified_img = magnify(preprocessed_img)
+
+    return preprocessed_img, preprocessed_img, magnified_img
     
 @spaces.GPU()
-def process(
+def magnify(
     input_image: Image.Image,
     user_prompt = "",
     positive_prompt = "clean, high-resolution, 8k, best quality, masterpiece",
@@ -323,14 +331,14 @@ with gr.Blocks(css=css, theme=theme) as demo:
                     inputs=[
                         input_image,
                     ],
-                    outputs=[result_gallery],
-                    fn=process,
+                    outputs=[preprocessed_image, result_gallery],
+                    fn=preprocess_n_magnify,
                     cache_examples=True,
                 )
         inputs = [
             input_image,
         ]
-        run_button.click(fn=process, inputs=preprocessed_image, outputs=[result_gallery])
+        run_button.click(fn=magnify, inputs=preprocessed_image, outputs=[result_gallery])
         input_image.upload(fn=preprocess_image,inputs=input_image, outputs=[preprocessed_image])
 
 demo.launch(share=True)