| import os |
| import warnings |
| from pathlib import Path |
|
|
| import gradio as gr |
| from deoldify import device |
| from deoldify.device_id import DeviceId |
| from deoldify.visualize import get_image_colorizer |
| from huggingface_hub import snapshot_download |
|
|
| os.system("pip freeze") |
| warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?") |
|
|
| device.set(device=DeviceId.CPU) |
|
|
| REPO_ID = "leonelhs/deoldify" |
| snapshot_folder = snapshot_download(repo_id=REPO_ID) |
|
|
| device.set(device=DeviceId.GPU0) |
| colorizer = get_image_colorizer(root_folder=Path(snapshot_folder), artistic=True) |
|
|
|
|
| def predict(image): |
| return colorizer.get_transformed_image(image, render_factor=35, watermarked=False) |
|
|
|
|
| title = "DeOldify" |
| description = r""" |
| ## Colorize image |
| |
| This is an implementation of <a href='https://github.com/jantic/DeOldify' target='_blank'>DeOldify</a>. |
| It has no any particular purpose than start research on AI models. |
| |
| """ |
|
|
| article = r""" |
| Questions, doubts, comments, please email 📧 `leonelhs@gmail.com` |
| |
| This demo is running on a CPU, if you like this project please make us a donation to run on a GPU or just give us a <a href='https://github.com/jantic/DeOldify' target='_blank'>Github ⭐</a> |
| |
| <a href="https://www.buymeacoffee.com/leonelhs"><img src="https://img.buymeacoffee.com/button-api/?text=Buy me a coffee&emoji=&slug=leonelhs&button_colour=FFDD00&font_colour=000000&font_family=Cookie&outline_colour=000000&coffee_colour=ffffff" /></a> |
| |
| <center><img src='https://visitor-badge.glitch.me/badge?page_id=deoldify.visitor-badge' alt='visitor badge'></center> |
| """ |
|
|
| demo = gr.Interface( |
| predict, [ |
| gr.Image(type="filepath", label="Image gray scale"), |
| ], [ |
| gr.Image(type="pil", label="Image color") |
| ], |
| title=title, |
| description=description, |
| article=article) |
|
|
| demo.queue().launch() |
|
|
