File size: 744 Bytes
d646931
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import gradio as gr
from utils import generate_sticker

def predict(image, prompt):
    result_img = generate_sticker(image, prompt)
    return result_img

with gr.Blocks() as demo:
    gr.Markdown("# 🦄 AI Sticker Generator (Stable Diffusion + IP-Adapter)")
    with gr.Row():
        image_input = gr.Image(type="pil", label="Upload your photo")
        prompt_input = gr.Textbox(
            label="Prompt (style or mood for emoji)",
            value="cartoon emoji, white outline, clean background",
        )
    output_image = gr.Image(label="Sticker Output")
    run_btn = gr.Button("Generate Sticker")
    run_btn.click(predict, inputs=[image_input, prompt_input], outputs=output_image)

if __name__ == "__main__":
    demo.launch()