Spaces:
Running
Running
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()
|