File size: 740 Bytes
ab03ae0 4d7123e ae69d05 4d7123e e8cb6ca 4d7123e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import gradio as gr
from transformers import pipeline
# Load the BigGAN model using the transformers pipeline
generator = pipeline("text-to-image", model="osanseviero/BigGAN-deep-128")
def generate_image(text_input):
"""Generates an image from text using the BigGAN pipeline."""
# The pipeline's output is an image.
image = generator(text_input)["images"][0]
return image
# Create the Gradio interface directly
interface = gr.Interface(
fn=generate_image,
inputs=gr.Textbox(label="Enter a prompt"),
outputs=gr.Image(label="Generated Image"),
title="BigGAN ImageNet",
description="BigGAN text-to-image demo.",
examples=[["american robin"], ["ocean sunset"], ["cat in a hat"]]
)
interface.launch()
|