Update app.py

app.py

@@ -1,31 +1,49 @@
 import gradio as gr
-from PIL import Image
+import cv2
 import numpy as np
+from PIL import Image
+import os
 import torch
 from basicsr.archs.rrdbnet_arch import RRDBNet
 from realesrgan import RealESRGANer
 
-model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64,
-                num_block=23, num_grow_ch=32, scale=4)
-
-upscaler = RealESRGANer(
+# Load Real-ESRGAN Model
+model = RealESRGANer(
     scale=4,
     model_path='RealESRGAN_x4plus.pth',
-    model=model,
-    tile=200,
+    model=RRDBNet(num_in_ch=3, num_out_ch=3, nf=64, nb=23, gc=32, scale=4, group=1, norm_type=None, act_type='leakyrelu', mode='NCHW', convtype='Conv'),
+    tile=0,
     tile_pad=10,
     pre_pad=0,
-    half=torch.cuda.is_available()
+    half=True,
+    gpu_id=None if not torch.cuda.is_available() else 0
 )
 
-def upscale_image(img):
+def upscale_image(input_image, scale, file_type):
     try:
-        img = np.array(img)
-        output, _ = upscaler.enhance(img, outscale=4)
-        return Image.fromarray(output)
+        img = np.array(input_image)
+        output, _ = model.enhance(img, outscale=scale)
+        output_pil = Image.fromarray(output)
+
+        output_path = "upscaled_output." + file_type.lower()
+        output_pil.save(output_path, dpi=(300, 300))
+
+        return output_path
     except Exception as e:
         return f"Error: {str(e)}"
 
-gr.Interface(fn=upscale_image, inputs=gr.Image(type="pil"), outputs=gr.Image(type="pil"),
-             title="AI Image Upscaler 4x",
-             description="Upload an image to upscale using RealESRGAN (4x)").launch()
+with gr.Blocks() as demo:
+    gr.Markdown("## 🖼️ AI Image Upscaler (Real-ESRGAN Powered)")
+
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Upload Image")
+        scale_input = gr.Radio([2, 4, 6, 8], label="Upscale Factor", value=4)
+        format_input = gr.Dropdown(["PNG", "JPEG", "TIFF"], label="Export Format", value="PNG")
+
+    output_image = gr.File(label="Download Upscaled Image")
+
+    upscale_btn = gr.Button("Upscale & Download")
+
+    upscale_btn.click(fn=upscale_image, inputs=[image_input, scale_input, format_input], outputs=output_image)
+
+demo.launch()