app.py

import os
import gradio as gr
import random
import util
from random import randint
import sys
from subprocess import call

is_colab = util.is_google_colab()

def run_cmd(command):
    try:
        print(command)
        call(command, shell=True)
    except KeyboardInterrupt:
        print("Process interrupted")
        sys.exit(1)

def inference(img):
    _id = randint(1, 10000)
    INPUT_DIR = "/tmp/input_image" + str(_id) + "/"
    OUTPUT_DIR = "/tmp/output_image" + str(_id) + "/"
    run_cmd("rm -rf " + INPUT_DIR)
    run_cmd("rm -rf " + OUTPUT_DIR)
    run_cmd("mkdir " + INPUT_DIR)
    run_cmd("mkdir " + OUTPUT_DIR)
    img.save(INPUT_DIR + "1.jpg", "JPEG")

    run_cmd("python test.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
    return os.path.join(OUTPUT_DIR, "1_out.jpg")

input_image = gr.inputs.Image(type="pil", label="Input")
output_image = gr.outputs.Image(type="file", label="Output")

demo = gr.Interface(
    inference,
    inputs=[input_image],
    outputs=[output_image]
)

demo.launch(debug=is_colab, share=is_colab)