app.py

import gradio as gr
from transformers import eBart

# Load the DALL-E Mega model
model = eBart.from_pretrained("dalle-mini/dalle-mega")

# Define the function to generate images
def generate_image(text):
    inputs = model.tokenizer(text, return_tensors="pt")
    outputs = model.generate(**inputs)
    image_url = model.tokenizer.decode(outputs[0], skip_special_tokens=True)
    return image_url

# Create the Gradio interface
ui = gr.Interface(
    fn=generate_image,
    inputs="text",
    outputs="image",
    title="DALL-E Mega Image Generator"
)

# Launch the Gradio app
ui.launch()