Spaces:
Running
Running
import gradio as gr | |
from utils import generate_sticker | |
def predict(image, prompt): | |
result_img = generate_sticker(image, prompt) | |
return result_img # Should be PIL Image or np.array or filepath | |
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(server_name="0.0.0.0", share=True) | |