fixed errors

app.py

@@ -21,6 +21,26 @@ model_ft  = Swin2SRForImageSuperResolution.from_pretrained(FINETUNED_ID, local_f
 model_pre.eval()
 model_ft.eval()
 
+
+ABOUT_TEXT = """
+    **About this tool**
+
+    This is a free image enhancement tool that uses Swin2SR to reconstruct and refine image detail at the pixel level (“pixel refabrication”).  
+    I built it because many of my own photos are distorted or only exist as low‑quality copies on social media.
+
+    **Tip for social media images**
+
+    If your only copy is on Instagram, open Instagram on a computer, view the photo in full‑screen, take a screenshot, and crop it tightly around the image before uploading here.
+
+    **Usage notes**
+
+    Processing and queue times can occasionally exceed ten minutes, especially during heavy use.  
+    If this tool gets regular traffic, I plan to upgrade to a stronger GPU backend.
+
+    Please reach out if you need any help or have questions.
+    """  
+
+
 # ---- Inference function ----
 def swin2sr_upscale(input_image: Image.Image, mode: str):
     """
@@ -29,7 +49,10 @@ def swin2sr_upscale(input_image: Image.Image, mode: str):
     """
     if input_image is None:
         return None
-
+    
+    # original size
+    w_lr, h_lr = input_image.size
+    
     if mode == "Smooth (fine-tuned)":
         model = model_ft
         processor = processor_ft
@@ -56,25 +79,9 @@ def swin2sr_upscale(input_image: Image.Image, mode: str):
         f"(scale factors: {w_sr / w_lr:.1f}×, {h_sr / h_lr:.1f}×)"
     )
 
-    ABOUT_TEXT = """
-    **About this tool**
-
-    This is a free image enhancement tool that uses Swin2SR to reconstruct and refine image detail at the pixel level (“pixel refabrication”).  
-    I built it because many of my own photos are distorted or only exist as low‑quality copies on social media.
-
-    **Tip for social media images**
-
-    If your only copy is on Instagram, open Instagram on a computer, view the photo in full‑screen, take a screenshot, and crop it tightly around the image before uploading here.
-
-    **Usage notes**
-
-    Processing and queue times can occasionally exceed ten minutes, especially during heavy use.  
-    If this tool gets regular traffic, I plan to upgrade to a stronger GPU backend.
-
-    Please reach out if you need any help or have questions.
-    """    
+  
 
-    return sr_image, msg, ABOUT_TEXT
+    return sr_image, msg
 
 # ---- Gradio UI ----
 with gr.Blocks() as demo:
@@ -105,7 +112,7 @@ with gr.Blocks() as demo:
     run_btn.click(
         fn=swin2sr_upscale,
         inputs=[input_image, mode_dropdown],
-        outputs=output_image,
+        outputs=[output_image, res_info]
     )
 
 if __name__ == "__main__":