Add models and logic

.gitattributes

@@ -0,0 +1,38 @@
+*.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
+NotoColorEmoji.ttf filter=lfs diff=lfs merge=lfs -text
+TwitterColorEmoji.ttf filter=lfs diff=lfs merge=lfs -text
+AppleColorEmoji.ttc filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+__pycache__
+.DS_Store

AppleColorEmoji.ttc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4be60f2801f40c651abd9c457b957114c88cc9385ac132356b588cc4a9ba2a00
+size 189791856

NotoColorEmoji.ttf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5074922e76b4fab18cb8dada2448d09a26ec2df978843ebe34aca29348f313df
+size 23709552

README.md

@@ -0,0 +1,12 @@
+---
+title: Bigmojis Upscaler
+emoji: ☝️
+colorFrom: yellow
+colorTo:  gray
+sdk: gradio
+sdk_version: 3.50.2
+app_file: app.py
+pinned: false
+license: mit
+---
+

TwitterColorEmoji.ttf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa0a9af2231807eebc35f9069ce9404019d1e10311cbf5d3c35277d6ece87463
+size 14477348

app.py

@@ -0,0 +1,191 @@
+#!/usr/bin/env python
+
+from __future__ import annotations
+
+import os
+import random
+
+import gradio as gr
+import numpy as np
+from PIL import Image, ImageDraw, ImageFont, ImageOps
+import uuid
+
+import argparse
+import cv2
+import glob
+import os
+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
+
+DESCRIPTION = '''<center><h1>☝️ Bigmojis ☝️</h1></span>  
+<span font-size:16px;">An emoji upscaler, for when you <i>really</i> mean it</span>  
+</center>
+
+Space by [ZachNagengast](https://huggingface.co/ZachNagengast) 
+
+[Follow me on Twitter!](https://twitter.com/ZachNagengast)
+
+Upscaler models provided by the [BasicSR](https://github.com/XPixelGroup/BasicSR) python package.
+
+Emojis by [Apple](https://www.apple.com).
+'''
+
+upsampler = None
+netscale = 4
+
+# load model
+def load_model(model_type='RealESRGAN_x4plus'):    
+    if model_type == 'RealESRGAN_x4plus':  # x4 RRDBNet model
+        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 = f'models/{model_type}.pth'
+    elif model_type == 'RealESRNet_x4plus':  # x4 RRDBNet model
+        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 = f'models/{model_type}.pth'
+    elif model_type == 'RealESRGAN_x4plus_anime_6B':  # x4 RRDBNet model with 6 blocks
+        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 = f'models/{model_type}.pth'
+    elif model_type == 'RealESRGAN_x2plus':  # x2 RRDBNet model
+        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 = f'models/{model_type}.pth'
+    elif model_type == 'realesr-animevideov3':  # x4 VGG-style model (XS size)
+        model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16, upscale=4, act_type='prelu')
+        netscale = 4
+        file_url = f'models/{model_type}.pth'
+
+    ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
+    model_path = os.path.join(ROOT_DIR, file_url)
+
+    upsampler = RealESRGANer(
+                    scale=netscale,
+                    model_path=model_path,
+                    model=model,
+                    
+                )
+    
+    return upsampler
+
+def switch_model(model_type):
+    global upsampler
+    upsampler = load_model(model_type)
+
+def save_image(img):
+    unique_name = str(uuid.uuid4()) + '.png'
+    img.save(unique_name)
+    return unique_name
+
+def generate_emoji_image(
+        prompt: str,
+        font_type: str = "Apple",
+        background: str = "white",
+        size: int = 160,
+        padding: int = 0,
+):
+    font_name = "AppleColorEmoji.ttc"
+    if font_type == "Google":
+        font_name = "NotoColorEmoji.ttf"
+    elif font_type == "Twitter":
+        font_name = "TwitterColorEmoji.ttf"
+    
+    font = ImageFont.truetype(
+        font_name, size=160
+    )
+
+    if background == "transparent":
+        background = (0, 0, 0, 0)
+
+    im = Image.new("RGBA", (size, size), background)
+    d = ImageDraw.Draw(im)
+
+    d.text((padding, padding), prompt, fill='white', embedded_color=True, font=font)
+    return im
+
+def generate_preview(
+    prompt: str,
+    font_type: str = "Apple",
+    background: str = "white",
+):
+    im = generate_emoji_image(prompt, font_type, background, 180, 10)
+    return im
+
+
+def upscale_image(img):
+    print(f"Upscaling...")
+    cv2_im = np.array(img)
+    output, _ = upsampler.enhance(cv2_im, outscale=netscale)
+    return Image.fromarray(output)
+
+def generate_upscaled_emoji(
+    prompt: str,
+    font_type: str = "Apple",
+    background: str = "white",
+    model: str = "RealESRGAN_x4plus",
+):
+    padding = 10 # prevents border artifacts
+    im = generate_emoji_image(prompt, font_type, background, 160+padding*2, padding)
+
+    result = upscale_image(im)
+
+    # crop padding
+    cropped = ImageOps.crop(result, border=padding)
+
+    return cropped
+
+
+
+examples = ['🤗', '😂', '🦙', '👍', '💸', '✨', '🚀', '🫱🏼‍🫲🏾', '🎉', '😎', '🛸', '🍩', '🦜', '🗿', '🧌', '🦋', '🆙']
+
+with gr.Blocks(css="style.css") as demo:
+    gr.Markdown(DESCRIPTION)
+    with gr.Row():
+        with gr.Column():
+            with gr.Group():
+                with gr.Row():
+                    prompt = gr.Text(
+                        label="Emoji",
+                        elem_id="prompt-text",
+                        show_label=True,
+                        max_lines=1,
+                        placeholder="Enter your emoji",
+                        container=False,
+                    )
+                    run_button = gr.Button("Upscale", scale=0)
+            with gr.Row():
+                preview = gr.Image(label="Glyph Bitmap", show_label=True, scale=1)
+                background = gr.Radio(choices=["transparent", "white", "black", "red", "green", "blue"], value="transparent", label="Background Color")
+                font_type = gr.Radio(choices=["Apple", "Google", "Twitter"], value="Apple", label="Type", visible=True)
+            with gr.Row():
+                gr.Examples(
+                    examples=examples,
+                    inputs=prompt,
+                    outputs=[preview],
+                    fn=generate_emoji_image,
+                    cache_examples=False,
+                )
+            with gr.Row():
+                model = gr.Radio(choices=['RealESRNet_x4plus','RealESRGAN_x4plus','RealESRGAN_x4plus_anime_6B','RealESRGAN_x2plus', 'realesr-animevideov3'], value="RealESRNet_x4plus", label="Model")
+        with gr.Column():
+            upscaled = gr.Image(label="Upscaled", value="huggingface-big.png")
+            gr.DuplicateButton(
+                value="Duplicate Space for private use",
+                elem_id="duplicate-button",
+                visible=True,
+            )
+
+    prompt.change(generate_preview, inputs=[prompt, font_type, background], outputs=[preview], api_name="run")
+    background.change(generate_preview, inputs=[prompt, font_type, background], outputs=[preview], api_name="run")
+    font_type.change(generate_preview, inputs=[prompt, font_type, background], outputs=[preview], api_name="run")
+    model.change(switch_model, inputs=[model], outputs=[])
+    prompt.submit(generate_upscaled_emoji, inputs=[prompt, font_type, background, model], outputs=[upscaled], api_name="run")
+    run_button.click(generate_upscaled_emoji, inputs=[prompt, font_type, background, model], outputs=[upscaled], api_name="run")
+
+if __name__ == "__main__":
+    upsampler = load_model('RealESRNet_x4plus')
+
+    demo.queue(max_size=20).launch()

models/RealESRGAN_x2plus.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:49fafd45f8fd7aa8d31ab2a22d14d91b536c34494a5cfe31eb5d89c2fa266abb
+size 67061725

models/RealESRGAN_x4plus.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fa0d38905f75ac06eb49a7951b426670021be3018265fd191d2125df9d682f1
+size 67040989

models/RealESRGAN_x4plus_anime_6B.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f872d837d3c90ed2e05227bed711af5671a6fd1c9f7d7e91c911a61f155e99da
+size 17938799

models/RealESRNet_x4plus.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a820b9bde89a874d7599d545567308ce6c128fc8754a53208eda016d40aa81df
+size 67040989

models/realesr-animevideov3.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8a8376811077954d82ca3fcf476f1ac3da3e8a68a4f4d71363008000a18b75d
+size 2504012

requirements.txt

@@ -0,0 +1,13 @@
+basicsr>=1.4.2
+facexlib>=0.2.5
+gfpgan>=1.3.5
+numpy
+opencv-python
+Pillow
+torch>=1.7
+torchvision
+transformers
+tqdm
+facexlib
+gfpgan
+basicsr

style.css

@@ -0,0 +1,4 @@
+h1 {
+    text-align: center;
+  }
+