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()