Update app.py

app.py

@@ -2,15 +2,11 @@ import gradio as gr
 from PIL import Image
 from transformers import pipeline
 
-# Use Hugging Face pre-trained ESRGAN model
-sr = pipeline("image-to-image", model="eugenesiow/esrgan")
+# Swin2SR super resolution model (x4 upscale)
+sr = pipeline("image-to-image", model="caidas/swin2SR-classical-sr-x4-64")
 
 def enhance(image: Image.Image):
-    # Run ESRGAN on the input
     results = sr(image)
-
-    # Hugging Face pipeline returns a list of dicts
-    # Example: [{'generated_image': <PIL.Image.Image>}]
     enhanced_img = results[0]["generated_image"]
     return enhanced_img
 
@@ -18,8 +14,8 @@ iface = gr.Interface(
     fn=enhance,
     inputs=gr.Image(type="pil", label="Upload Photo"),
     outputs=gr.Image(type="pil", label="Enhanced Photo"),
-    title="Free AI Photo Enhancer",
-    description="Upload your photo and download the enhanced version (AI Super Resolution)."
+    title="AI Photo Enhancer",
+    description="Upload your photo and get an enhanced version (Super Resolution x4)."
 )
 
 iface.launch()