import gradio as gr
from transformers import pipeline
from PIL import Image

# Load the image-to-image model from Hugging Face
model = pipeline("image-to-image")

def generate_ai_image(image):
    # Convert the image to an AI-generated image
    ai_image = model(image)[0]
    return ai_image

# Create the Gradio interface
iface = gr.Interface(
    fn=generate_ai_image,
    inputs=gr.Image(type="pil"),
    outputs=gr.Image(type="pil"),
    title="AI Image Generator",
    description="Upload a normal picture and convert it to an AI-generated image."
)

# Run the app
if __name__ == "__main__":
    iface.launch()