File size: 617 Bytes
b7be5ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
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()
|