Text-to-Image / app.py
LuigiLui's picture
Update app.py
25ee7c1 verified
raw
history blame
2.29 kB
import gradio as gr
import random
model = gr.load("models/Purz/face-projection")
def generate_image(text, seed):
if seed is not None:
random.seed(seed)
return model(text)
examples = [
["Humanoid Cat Warrior, Full View", None],
["Warhammer Sisterhood", None],
["Future Robots war", None],
["Fantasy dragon", None]
]
# Custom CSS for minimalist style, horizontal example layout, and styled prompt box
custom_css = """
/* Prompt textbox styling */
input[type="text"] {
border: 2px solid #ddd; /* Default border */
border-radius: 8px;
padding: 12px;
font-size: 16px;
width: 100%;
background-color: #f94925; /* Light gray background */
}
/* Examples arranged in a single line, without buttons */
.gr-examples {
display: flex;
flex-direction: row;
justify-content: space-evenly;
gap: 10px;
margin-top: 10px;
}
.gr-examples span {
font-size: 14px;
color: #333;
cursor: pointer;
padding: 5px;
background-color: white; /* Simple white background */
border: 1px solid #ddd; /* Light gray border */
border-radius: 5px; /* Rounded corners */
transition: all 0.2s ease;
}
/* Hover effect on examples */
.gr-examples span:hover {
background-color: #000000; /* Light gray on hover */
color: #007bff; /* Change text color on hover */
text-decoration: underline; /* Underline text on hover */
}
/* Styling for the generate button */
button {
border-radius: 50px;
background: linear-gradient(90deg, #ff8a00, #e52e71);
color: white;
font-weight: bold;
padding: 10px 20px;
border: none;
cursor: pointer;
font-size: 16px;
margin-top: 10px;
}
button:hover {
background: #fafafa;
}
"""
# Gradio Interface with updated UI elements and custom CSS
interface = gr.Interface(
fn=generate_image,
inputs=[
gr.Textbox(label="Imagine Your Dream:", placeholder="Describe the scene...", interactive=True),
gr.Slider(minimum=0, maximum=10000, step=1, label="Seed (optional)")
],
outputs=gr.Image(label="Generated Image", type="pil"),
examples=examples,
css=custom_css, # Apply custom CSS
description="Enter a creative prompt or choose an example to generate your AI-powered image.",
)
interface.launch()