Create app.py

app.py

@@ -0,0 +1,37 @@
+from base64 import b64encode, b64decode
+from io import BytesIO
+from pathlib import Path
+
+import numpy as np
+from basicsr.archs.rrdbnet_arch import RRDBNet
+from PIL import Image
+from realesrgan import RealESRGANer
+from huggingface_hub import hf_hub_download
+import gradio as gr
+
+model = RRDBNet(num_in_ch=3, num_out_ch=3)
+upsampler = RealESRGANer(
+    scale=4,
+    model_path=hf_hub_download('nateraw/real-esrgan', 'RealESRGAN_x4plus.pth'),
+    model=model,
+    tile=0,
+    pre_pad=0,
+    half=True,
+)
+
+def upsample(image):
+    # image = np.array(image)
+    image = image[:, :, ::-1]  # RGB -> BGR
+    image, _ = self.upsampler.enhance(image, outscale=4)
+    image = image[:, :, ::-1]  # BGR -> RGB
+    image = Image.fromarray(image)
+    return image
+
+interface = gr.Interface(
+    upsample,
+    inputs=["image"],
+    outputs=["image"]
+)
+
+if __name__ == '__main__':
+    interface.launch()