| from transformers import pipeline | |
| import gradio as gr | |
| generator = pipeline("image-to-image", model="timbrooks/instruct-pix2pix") | |
| def generate_face_variations(image): | |
| variations = generator(image, num_images=4, guidance_scale=7.5) | |
| return variations | |
| with gr.Blocks() as demo: | |
| gr.Markdown("# Face Position Variation Generator") | |
| with gr.Row(): | |
| input_image = gr.Image(type="pil") | |
| output_gallery = gr.Gallery().style(grid=2, height="auto") | |
| generate_btn = gr.Button("Generate") | |
| generate_btn.click(fn=generate_face_variations, inputs=input_image, outputs=output_gallery) | |
| demo.launch() | |