| import gradio as gr | |
| import random | |
| model = gr.load("models/Purz/face-projection") | |
| def generate_image(text, seed, style): | |
| if seed is not None: | |
| random.seed(seed) | |
| if text in [example[0] for example in examples]: | |
| print(f"Using example: {text}") | |
| if style == "Anime": | |
| text = f"Anime style, {text}" | |
| elif style == "Realistic": | |
| text = f"Realistic style, {text}" | |
| elif style == "Photographic": | |
| text = f"Photographic style, {text}" | |
| return model(text) | |
| examples = [ | |
| ["Humanoid Cat Warrior, Full View", None], | |
| ["Warhammer Sisterhood", None], | |
| ["Future Robots war", None], | |
| ["Fantasy dragon", None] | |
| ] | |
| interface = gr.Interface( | |
| fn=generate_image, | |
| inputs=[ | |
| gr.Textbox(label="Type here your imagination:", placeholder="Type or click an example..."), | |
| gr.Slider(minimum=0, maximum=10000, step=1, label="Seed (optional)"), | |
| gr.Radio(choices=["Anime", "Realistic", "Photographic"], label="Style", value="Realistic") | |
| ], | |
| outputs=gr.Image(label="Generated Image"), | |
| examples=examples, | |
| theme="NoCrypt/miku", | |
| description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.", | |
| ) | |
| interface.launch() |