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