Update app.py

app.py

@@ -1,7 +1,47 @@
 import gradio as gr
+import cv2
+import numpy as np
+from PIL import Image
+from basicsr.archs.rrdbnet_arch import RRDBNet
 
-def greet(name):
-    return "Hello " + name + "!!"
+from realesrgan import RealESRGANer
+from realesrgan.archs.srvgg_arch import SRVGGNetCompact
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+def upscale(image: Image, model: str, outscale: int = 4, name: str = None):
+    if model.startswith("https://"):
+        model_name = name
+    else:
+        model_name = model.split('.')[0]
+    if model_name in ['RealESRGAN_x4plus', 'RealESRNet_x4plus']:  # x4 RRDBNet model
+        m = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)
+        netscale = 4
+    elif model_name in ['RealESRGAN_x4plus_anime_6B']:  # x4 RRDBNet model with 6 blocks
+        m = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=6, num_grow_ch=32, scale=4)
+        netscale = 4
+    elif model_name in ['RealESRGAN_x2plus']:  # x2 RRDBNet model
+        m = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=2)
+        netscale = 2
+    elif model_name in ['realesr-animevideov3']:  # x4 VGG-style model (XS size)
+        m = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16, upscale=4, act_type='prelu')
+        netscale = 4
+
+    upsampler = RealESRGANer(
+        scale=netscale, model_path=model, model=m, device='cpu', half=False, tile=0, tile_pad=10, pre_pad=0)
+
+    img = cv2.imread(image, cv2.IMREAD_UNCHANGED)
+
+    try:
+        output = upsampler.enhance(img, outscale=outscale)
+    except Exception as e:
+        print(e)
+        return None
+    else:
+        return np.asarray(Image.fromarray(cv2.cvtColor(output, cv2.COLOR_BGR2RGB)))
+
+
+gr.Interface(
+    upscale, [gr.Image(type="pil"), 'text', 'text', 'text'],
+    gr.Image(type="pil", label="Output"),
+    title='hi',
+    description='desc').launch()