birthdayCrown / wear_birthday_crown.py
liangtian's picture
Update wear_birthday_crown.py
e96083d
import os
import cv2
import numpy as np
import gradio as gr
import PIL
from PIL import Image
def exec_cmd(image):
image.save("me.jpg")
os.system("python create_gif.py --config config.json --image me.jpg --output result.gif")
output="result.gif"
return output
interface = gr.Interface(fn=exec_cmd, inputs=gr.inputs.Image(type="pil"),outputs=gr.outputs.Image(type="file"))
interface.launch()