Update app.py

app.py

@@ -11,14 +11,19 @@ from transformers import AutoProcessor, AutoModelForCausalLM
 from gradio_imageslider import ImageSlider
 from PIL import Image
 from huggingface_hub import snapshot_download
-import requests# For ESRGAN (requires pip install basicsr gfpgan)
+import requests
+import io
+
+# For ESRGAN (requires pip install basicsr gfpgan)
 try:
     from basicsr.archs.rrdbnet_arch import RRDBNet
     from basicsr.utils import img2tensor, tensor2img
     USE_ESRGAN = True
 except ImportError:
     USE_ESRGAN = False
-    warnings.warn("basicsr not installed; falling back to LANCZOS interpolation.")css = """
+    warnings.warn("basicsr not installed; falling back to LANCZOS interpolation.")
+
+css = """
 #col-container {
     margin: 0 auto;
     max-width: 800px;
@@ -27,7 +32,9 @@ except ImportError:
     text-align: center;
     margin-bottom: 2rem;
 }
-"""# Device setup
+"""
+
+# Device setup
 if torch.cuda.is_available():
     power_device = "GPU"
     device = "cuda"