FaceOver / app.py
AmitIsraeli's picture
change title
fd2de02
raw
history blame
1.13 kB
import gradio as gr
from help_function import help_function
from PIL import Image
import numpy as np
model_helper = help_function()
def greet(image,text,power):
PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
image_edit = model_helper.image_from_text(text,PIL_image,power)
return image_edit
description = "demo for model to edit face with text, you can see the [github repo CelebrityLook](https://github.com/amit154154/CelebrityLook)"
title = "FaceOver - edit face with text 🐨 "
examples = [
["examples/jennifer lawrence.jpg", "dark hair smiling woman with glasses", 0.58],
["examples/kim kardashian.jpg", "beautiful asian woman with short haired ", 0.88],
["examples/Arnold Schwarzenegger.jpg", "happy man with long hair and hat", 0.44]
]
iface = gr.Interface(fn=greet,
inputs=["image", "text", gr.inputs.Slider(0.0, 1.0)],
outputs="image",
title=title,
description=description,
cache_examples=False,
examples=examples
)
iface.launch()