Add files using upload-large-folder tool

Ilaria_Upscaler/.gitattributes

@@ -0,0 +1,36 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm 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
+ilariaaisuite.png filter=lfs diff=lfs merge=lfs -text

Ilaria_Upscaler/LICENSE.md

@@ -0,0 +1,32 @@
+# Ilaria's Non-Commercial Use License (INCU)
+
+## 1. Definitions
+
+- "Software" refers to the content present within this repository including code, graphics, sounds, images, text, and other files.
+- "You" refers to the individual or entity that wishes to use the Software.
+- "Ilaria" refers to the original creator of the Software.
+- "Fork" refers to the action of creating a personal copy of another user's repository.
+
+## 2. Grant of Rights
+
+Subject to the terms of this license, Ilaria hereby grants You a worldwide, royalty-free, non-exclusive, perpetual license to use the Software for personal, non-commercial purposes. This includes the rights to use, copy, modify, merge, publish, and distribute the Software for non-commercial purposes.
+
+## 3. Redistribution
+
+You may not distribute or sell the Software, or any derivative works based on the Software, unless you have been specifically granted permission by Ilaria. Any permitted redistribution must also be under the terms of this license. Unauthorized distribution is strictly prohibited and will result in the termination of this license.
+
+## 4. Commercial Use
+
+Commercial use of the Software is strictly prohibited without the express written consent of Ilaria. Small businesses may be granted permission to use the Software for profit, but large corporations must negotiate a commercial agreement with Ilaria. Unauthorized commercial use is strictly prohibited and will result in the termination of this license.
+
+## 5. Forking and Personal Use
+
+You are free to fork and modify the Software for your own personal use. You may not distribute, publicly display, or create derivative works from your modifications unless granted permission by Ilaria. All modifications must also be under the terms of this license.
+
+## 6. Liability
+
+Ilaria provides the Software "as is," without warranty of any kind, express or implied, including but not limited to the warranties of merchantability, fitness for a particular purpose, and non-infringement. In no event shall Ilaria be liable for any claim, damages, or other liability, whether in an action of contract, tort, or otherwise, arising from, out of, or in connection with the Software or the use or other dealings in the Software. You agree to use the Software at your own risk.
+
+## 7. Governing Law
+
+This license is governed by the laws of the jurisdiction in which Ilaria resides. Any disputes related to this license will be resolved in the courts of that jurisdiction.

Ilaria_Upscaler/README.md

@@ -0,0 +1,74 @@
+---
+title: Ilaria Upscaler
+emoji: 🌖
+colorFrom: pink
+colorTo: pink
+sdk: gradio
+sdk_version: 4.36.1
+app_file: app.py
+pinned: true
+---
+
+![Ilaria AI Suite](./ilariaaisuite.png)
+***
+[![Static Badge](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Space-s?labelColor=YELLOW&color=FFEA00)](https://huggingface.co/spaces/TheStinger/Ilaria_Upscaler) [![Static Badge](https://img.shields.io/badge/%F0%9F%A4%97%20HF%20Space-Duplication-s?labelColor=YELLOW&color=FFEA00)](https://huggingface.co/spaces/TheStinger/Ilaria_Upscaler?duplicate=true) [![Static Badge](https://img.shields.io/badge/GitHub-Source%20Code-s?logo=GitHub)](https://github.com/TheStingerX/Ilaria-Upscaler) [![Static Badge](https://img.shields.io/badge/AI%20Hub-Discord%20Server-s?logo=Discord&color=%09%237289da)](https://discord.gg/aihub) [![Static Badge](https://img.shields.io/badge/Ko--Fi-s?logo=Ko-Fi&label=Support%20me%20on&labelColor=434b57&color=FF5E5B)](https://ko-fi.com/ilariaowo)
+***
+<p align="center">
+  <h1>Ilaria Upscaler 💖</h1>
+</p>
+
+🎉 Welcome to Ilaria Upscaler! 🎉  
+  
+This project leverages various libraries and modules to create a Graphical User Interface (GUI) for upscaling images.  
+It's primarily designed for use with HuggingFace Spaces. 🤗   
+
+Ilaria Upscaler is part of the Ilaria AI Suite wich includes various easy and powerful tools. 💖
+
+## 📦 Installation 📦
+
+To use this project, clone the original Space on Hugging Face.  
+Make sure you restart it time to time to keep up with the new updates.
+
+## 🖥️ Usage 🖥️
+
+Once the dependencies are installed automatically, Hugging Face will use app.py to start the user interface.  
+From there, you can utilize the various features of the project.
+
+## 🌟 Features 🌟
+
+Ilaria Upscaler offers a range of features, including:
+
+- 🖼️ **Image Upscaling and Restoration**:  
+Ilaria Upscaler uses the RealESRGANer class from the realesrgan module to upscale and restore images.
+It supports multiple models, each with different scales and architectures, providing flexibility in the upscaling process.
+
+- 🎛️ **Adjustable Parameters**:  
+The realesrgan function allows users to adjust various parameters such as the model name, denoise strength, face enhancement, and output scale.
+This provides a high level of customization to fit different use cases.
+
+- 🖥️ **User-Friendly Web Interface**:  
+Ilaria Upscaler The code uses the gradio module to create an intuitive web interface.
+Users can easily upload an image, and the application will return the upscaled and restored image.
+
+- 📂 **Efficient File Handling**:  
+The code uses the os and basicsr.utils.download_util modules for efficient file handling operations.
+It checks if the model file exists in the local directory, and if not, it downloads the file from the specified URL.
+
+- 👩 **Face Enhancement Option**:  
+If the face_enhance parameter is set to True, the code uses the GFPGANer class from the gfpgan module to enhance faces in the image.  
+This is particularly useful for images with faces that need to be upscaled and enhanced separately from the rest of the image.
+
+## 🙏 Credits 🙏
+
+- **Squishy** - For helping me with the ideas
+
+## 🤝 Contributing 🤝
+
+Interested in contributing to this project? Ilaria is always looking for collaborators.  
+Feel free to open a pull request on Hugging Face.
+
+## 📄 License 📄
+
+This project is released under the `INCU` license.  
+For more details, please check the license file.  
+For further questions feel free to contact Ilaria.

Ilaria_Upscaler/app.py

@@ -0,0 +1,195 @@
+import gradio as gr
+import cv2
+import numpy
+import os
+import random
+from basicsr.archs.rrdbnet_arch import RRDBNet
+from basicsr.utils.download_util import load_file_from_url
+from realesrgan import RealESRGANer
+from realesrgan.archs.srvgg_arch import SRVGGNetCompact
+from torchvision.transforms.functional import rgb_to_grayscale
+import spaces
+
+last_file = None
+img_mode = "RGBA"
+
+@spaces.GPU
+def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
+    """Real-ESRGAN function to restore (and upscale) images."""
+    if not img:
+        return
+
+    # Define model parameters
+    if model_name == 'RealESRGAN_x4plus':
+        model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)
+        netscale = 4
+        file_url = ['https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth']
+    elif model_name == 'RealESRNet_x4plus':
+        model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)
+        netscale = 4
+        file_url = ['https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/RealESRNet_x4plus.pth']
+    elif model_name == 'RealESRGAN_x4plus_anime_6B':
+        model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=6, num_grow_ch=32, scale=4)
+        netscale = 4
+        file_url = ['https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth']
+    elif model_name == 'RealESRGAN_x2plus':
+        model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=2)
+        netscale = 2
+        file_url = ['https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth']
+    elif model_name == 'realesr-general-x4v3':
+        model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
+        netscale = 4
+        file_url = [
+            'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-wdn-x4v3.pth',
+            'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth'
+        ]
+
+    model_path = os.path.join('weights', model_name + '.pth')
+    if not os.path.isfile(model_path):
+        ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
+        for url in file_url:
+            model_path = load_file_from_url(
+                url=url, model_dir=os.path.join(ROOT_DIR, 'weights'), progress=True, file_name=None)
+
+    dni_weight = None
+    if model_name == 'realesr-general-x4v3' and denoise_strength != 1:
+        wdn_model_path = model_path.replace('realesr-general-x4v3', 'realesr-general-wdn-x4v3')
+        model_path = [model_path, wdn_model_path]
+        dni_weight = [denoise_strength, 1 - denoise_strength]
+
+    upsampler = RealESRGANer(
+        scale=netscale,
+        model_path=model_path,
+        dni_weight=dni_weight,
+        model=model,
+        tile=0,
+        tile_pad=10,
+        pre_pad=10,
+        half=False,
+        gpu_id=None
+    )
+
+    if face_enhance:
+        from gfpgan import GFPGANer
+        face_enhancer = GFPGANer(
+            model_path='https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth',
+            upscale=outscale,
+            arch='clean',
+            channel_multiplier=2,
+            bg_upsampler=upsampler)
+
+    cv_img = numpy.array(img)
+    img = cv2.cvtColor(cv_img, cv2.COLOR_RGBA2BGRA)
+
+    try:
+        if face_enhance:
+            _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True)
+        else:
+            output, _ = upsampler.enhance(img, outscale=outscale)
+    except RuntimeError as error:
+        print('Error', error)
+        print('If you encounter CUDA out of memory, try to set --tile with a smaller number.')
+    else:
+        extension = 'png' if img_mode == 'RGBA' else 'jpg'
+
+        out_filename = f"output_{rnd_string(8)}.{extension}"
+        cv2.imwrite(out_filename, output)
+        global last_file
+        last_file = out_filename
+
+        output_img = cv2.cvtColor(output, cv2.COLOR_BGRA2RGBA) if img_mode == "RGBA" else output
+        return out_filename, image_properties(output_img)
+
+def rnd_string(x):
+    characters = "abcdefghijklmnopqrstuvwxyz_0123456789"
+    return "".join((random.choice(characters)) for i in range(x))
+
+def reset():
+    global last_file
+    if last_file:
+        print(f"Deleting {last_file} ...")
+        os.remove(last_file)
+        last_file = None
+    return gr.update(value=None), gr.update(value=None), gr.update(value=None)
+
+def has_transparency(img):
+    if img.info.get("transparency", None) is not None:
+        return True
+    if img.mode == "P":
+        transparent = img.info.get("transparency", -1)
+        for _, index in img.getcolors():
+            if index == transparent:
+                return True
+    elif img.mode == "RGBA":
+        extrema = img.getextrema()
+        if extrema[3][0] < 255:
+            return True
+    return False
+
+def image_properties(img):
+    """Returns the dimensions (width and height) and color mode of the input image and
+    also sets the global img_mode variable to be used by the realesrgan function
+    """
+    global img_mode
+    if img is None:  # Explicitly check for None
+        return "No image data available."
+
+    if isinstance(img, numpy.ndarray):  # Handle NumPy array case
+        height, width = img.shape[:2]
+        channels = img.shape[2] if len(img.shape) > 2 else 1
+        img_mode = "RGBA" if channels == 4 else "RGB" if channels == 3 else "Grayscale"
+        return f"Resolution: Width: {width}, Height: {height}  |  Color Mode: {img_mode}"
+    
+    if hasattr(img, "info") and hasattr(img, "mode") and hasattr(img, "size"):  # Handle PIL images
+        if has_transparency(img):
+            img_mode = "RGBA"
+        else:
+            img_mode = "RGB"
+        return f"Resolution: Width: {img.size[0]}, Height: {img.size[1]}  |  Color Mode: {img_mode}"
+    
+    return "Unsupported image format."
+
+def main():
+    with gr.Blocks(theme=gr.themes.Default(primary_hue="pink", secondary_hue="rose"), title="Ilaria Upscaler 💖") as app:
+
+        gr.Markdown(
+            """# <div align="center"> Ilaria Upscaler 💖 </div>  
+        """
+        )
+        with gr.Accordion("Upscaling option"):
+            with gr.Row():
+                model_name = gr.Dropdown(label="Model", 
+                                        choices=["RealESRGAN_x4plus", "RealESRNet_x4plus", "RealESRGAN_x4plus_anime_6B", "RealESRGAN_x2plus", "realesr-general-x4v3"], 
+                                        value="RealESRGAN_x4plus")
+                denoise_strength = gr.Slider(label="Denoise Strength", minimum=0, maximum=1, step=0.1, value=0.5)
+                outscale = gr.Slider(label="Resolution Upscale", minimum=1, maximum=6, step=1, value=4)
+                face_enhance = gr.Checkbox(label="Face Enhancement")
+
+        with gr.Row():
+            with gr.Group():
+                input_image = gr.Image(label="Input Image", type="pil")
+                input_properties = gr.Textbox(label="Input Image Properties", interactive=False)
+
+            with gr.Group():
+                output_image = gr.Image(label="Output Image")
+                output_properties = gr.Textbox(label="Output Image Properties", interactive=False)
+
+        with gr.Row():
+            reset_btn = gr.Button("Reset")
+            upscale_btn = gr.Button("Upscale")
+
+        input_image.change(fn=image_properties, inputs=input_image, outputs=input_properties)
+        upscale_btn.click(fn=realesrgan, 
+                        inputs=[input_image, model_name, denoise_strength, face_enhance, outscale], 
+                        outputs=[output_image, output_properties])
+        reset_btn.click(fn=reset, inputs=[], outputs=[input_image, output_image, input_properties])
+
+        gr.Markdown(
+            """Made with love by Ilaria 💖 | Support me on [Ko-Fi](https://ko-fi.com/ilariaowo) | Using [Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN).
+        """
+        )
+
+    app.launch()
+
+if __name__ == "__main__":
+    main()

Ilaria_Upscaler/requirements.txt

@@ -0,0 +1,11 @@
+torch
+torchvision
+numpy
+opencv-python
+Pillow
+basicsr @ git+https://github.com/XPixelGroup/BasicSR.git
+facexlib
+gfpgan
+tqdm
+gradio
+realesrgan