update models

.gitattributes

@@ -32,4 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.xz filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,64 @@
+import gradio as gr
+import spaces
+from gradio_imageslider import ImageSlider
+from image_gen_aux import UpscaleWithModel
+from image_gen_aux.utils import load_image
+
+MODELS = {
+    "4xRealESRGAN": "luca115/4xRealESRGAN",
+    "4xUltraSharp": "Kim2091/UltraSharp",
+    "4xNomosWebPhoto_RealPLKSR": "OzzyGT/4xNomosWebPhoto_RealPLKSR",
+    "4xRemacri": "OzzyGT/4xRemacri",
+}
+
+
+@spaces.GPU
+def upscale_image(image, model_selection):
+    original = load_image(image)
+
+    upscaler = UpscaleWithModel.from_pretrained(MODELS[model_selection]).to("cuda")
+    image = upscaler(original, tiling=True, tile_width=1024, tile_height=1024)
+
+    return original, image
+
+
+def clear_result():
+    return gr.update(value=None)
+
+
+title = """<h1 align="center">Best Upscaling Models</h1>
+<div align="center">This space is a showcase of the different super resolution models you can use to upscale with the 
+<a href="https://github.com/asomoza/image_gen_aux">Image Generation Auxiliary Tools</a> library.</div>
+"""
+
+with gr.Blocks() as demo:
+    gr.HTML(title)
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="pil", label="Input Image")
+
+            model_selection = gr.Dropdown(
+                choices=list(MODELS.keys()),
+                value="UltraSharp",
+                label="Model",
+            )
+
+            run_button = gr.Button("Upscale")
+        with gr.Column():
+            result = ImageSlider(
+                interactive=False,
+                label="Generated Image",
+            )
+
+    run_button.click(
+        fn=clear_result,
+        inputs=None,
+        outputs=result,
+    ).then(
+        fn=upscale_image,
+        inputs=[input_image, model_selection],
+        outputs=result,
+    )
+
+
+demo.launch(share=False)

requirements.txt

@@ -0,0 +1,6 @@
+torch
+git+https://github.com/asomoza/image_gen_aux.git
+spaces
+gradio
+gradio-imageslider
+numpy==1.26.4