Gradio is a Python library for wrapping any ML model in a web interface, ready to deploy and scale as an app. This guide builds a GFPGAN-powered face-restoration demo with Gradio on Ubuntu 22.04, runs it as a systemd service, and exposes it through Nginx with TLS.
Prerequisites:a GPU-enabled Ubuntu 22.04 server, a domain A record (e.g.gradio.example.com
), non-root sudo access, Nginx installed.
1. Install dependencies:
$ pip3 install realesrgan gfpgan basicsr gradio
realesrgan
— background restorationgfpgan
— face restorationbasicsr
— provides RRDBNet
, the super-resolution architecture GFPGAN relies ongradio
— the web interface2. GFPGAN's pandas dependency needs jinja2 >= 3.1.2:
$ pip show jinja2
Upgrade if it's older:
$ pip install --upgrade jinja2
3. Create the project directory:
$ sudo mkdir -p /opt/gradio-webapp/
$ sudo chown -R :$(id -gn) /opt/gradio-webapp/
$ sudo chmod -R 775 /opt/gradio-webapp/
Uploads a face image and returns two enhanced outputs.
$ cd /opt/gradio-webapp/
$ nano app.py
python
import gradio as gr
from gfpgan import GFPGANer
from basicsr.archs.rrdbnet_arch import RRDBNet
from realesrgan import RealESRGANer
import numpy as np
import cv2
import requests
def enhance_image(input_image):
arch = 'clean'
model_name = 'GFPGANv1.4'
gfpgan_checkpoint = 'https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth'
realersgan_checkpoint = 'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth'
rrdbnet = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=2)
bg_upsampler = RealESRGANer(
scale=2,
model_path=realersgan_checkpoint,
model=rrdbnet,
tile=400,
tile_pad=10,
pre_pad=0,
half=True
)
restorer = GFPGANer(
model_path=gfpgan_checkpoint,
upscale=2,
arch=arch,
channel_multiplier=2,
bg_upsampler=bg_upsampler
)
input_image = input_image.astype(np.uint8)
cropped_faces, restored_faces, restored_img = restorer.enhance(input_image)
return restored_faces[0], restored_img
interface = gr.Interface(
fn=enhance_image,
inputs=gr.Image(),
outputs=[gr.Image(), gr.Image()],
live=True,
title="Face Enhancement with GFPGAN",
description="Upload an image of a face and see it enhanced using GFPGAN. Two outputs will be displayed: restored_faces and restored_img."
)
interface.launch(server_name="0.0.0.0", server_port=8080)
enhance_image()
loads the GFPGAN/Real-ESRGAN checkpoints and runs restoration; interface
wires that function to a Gradio UI listening on port 8080.
Test it:
$ python3 app.py
Running on local URL: http://0.0.0.0:8080
Set share=True
in launch()
for a temporary public Gradio link. Stop it with Ctrl+C once verified.
$ sudo nano /etc/systemd/system/my_gradio_app.service
Replace example-user
with your actual account:
[Unit]
Description=My Gradio Web Application
[Service]
ExecStart=/usr/bin/python3 /opt/gradio-webapp/app.py
WorkingDirectory=/opt/gradio-webapp/
Restart=always
User=example-user
Environment=PATH=/usr/bin:/usr/local/bin
Environment=PYTHONUNBUFFERED=1
[Install]
WantedBy=multi-user.target
bash
$ sudo systemctl daemon-reload
$ sudo systemctl enable my_gradio_app
$ sudo systemctl start my_gradio_app
$ sudo systemctl status my_gradio_app
bash
$ sudo nano /etc/nginx/sites-available/gradio.conf
server {
listen 80;
listen [::]:80;
server_name gradio.example.com;
location / {
proxy_pass http://127.0.0.1:8080/;
}
}
bash
$ sudo ln -s /etc/nginx/sites-available/gradio.conf /etc/nginx/sites-enabled/
$ sudo nginx -t
$ sudo systemctl restart nginx
bash
$ sudo ufw status
$ sudo ufw allow 80/tcp
$ sudo ufw allow 443/tcp
$ sudo ufw reload
$ sudo apt install -y certbot python3-certbot-nginx
$ sudo certbot --nginx -d gradio.example.com -m admin@example.com --agree-tos
$ sudo certbot renew --dry-run
Visit https://gradio.example.com
, upload a sample face image, and confirm you get back the restored face crop and the full restored image.
The Gradio app is running as a managed systemd service behind Nginx with TLS. From here:
interface.launch()
pattern works for any function-wrapped modelinterface.launch(auth=...)
) if the app shouldn't be fully publicFor the full guide, visit the original article on ** Vultr Docs**.