added output to be jpeg

app.py

@@ -46,24 +46,27 @@ def swin2sr_upscale(input_image: Image.Image, mode: str):
     sr_array = (sr_tensor.mul(255).byte().cpu().permute(1, 2, 0).numpy())
     sr_image = Image.fromarray(sr_array)
 
+    # ---ensure JPEG-compatible mode ---
+    sr_image = sr_image.convert("RGB")
+
     return sr_image
 
 # ---- Gradio UI ----
 with gr.Blocks() as demo:
     gr.Markdown("# Image Super-Resolution (Swin2SR x4)")
     gr.Markdown(
-        "Choose **Crystal clear (pretrained)** for the original Swin2SR model, "
+        "Choose **Clear (pretrained)** for the original Swin2SR model, "
         "or **Smooth (fine-tuned)** for the Swin2SR version we fine-tuned on DIV2K patches."
     )
 
     with gr.Row():
         input_image = gr.Image(type="pil", label="Upload low-res image")
-        output_image = gr.Image(type="pil", label="4x Super-resolved image")
+        output_image = gr.Image(type="pil",format="jpeg", label="4x Super-resolved image")
 
     mode_dropdown = gr.Dropdown(
         label="Style",
-        choices=["Crystal clear (pretrained)", "Smooth (fine-tuned)"],
-        value="Crystal clear (pretrained)",
+        choices=["Clear (OFtB)", "Smooth (Over Smoothing | tuned_1)"],
+        value="Clear (OFtB)",
         interactive=True,
     )