Create app.py

app.py

@@ -0,0 +1,33 @@
+app.pyimport gradio as gr
+import cv2
+from realesrgan import RealESRGANer
+
+# Khởi tạo mô hình Real-ESRGAN
+model_path = "weights/RealESRGAN_x4plus.pth"  # Đảm bảo đã tải mô hình
+upscaler = RealESRGANer(
+    scale=4,
+    model_path=model_path,
+    model=None,
+    tile=0,
+    tile_pad=10,
+    pre_pad=0,
+    half=True,  # Sử dụng FP16 nếu có GPU
+)
+
+def upscale_image(image):
+    try:
+        output, _ = upscaler.enhance(image, outscale=4)
+        return output
+    except Exception as e:
+        return str(e)
+
+# Giao diện Gradio
+iface = gr.Interface(
+    fn=upscale_image,
+    inputs=gr.Image(type="numpy", label="Upload Image"),
+    outputs=gr.Image(type="numpy", label="Upscaled Image"),
+    title="Real-ESRGAN Upscaler",
+    description="Upload an image to upscale it using Real-ESRGAN."
+)
+
+iface.launch()