app.py

import gradio as gr
from transformers import pipeline

# Carregar o pipeline de geração de imagens
pipe = pipeline("image-generation", model="stabilityai/stable-diffusion-2-1")

def generate_image(prompt):
    """
    Gera uma imagem com base em um prompt textual usando o modelo.
    """
    try:
        # Gera a imagem
        image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)[0]["image"]
        return image
    except Exception as e:
        return f"Erro ao gerar imagem: {str(e)}"

# Configurar a interface do Gradio
with gr.Blocks() as demo:
    gr.Markdown("## Gerador de Imagens com Texto - Stable Diffusion")
    
    with gr.Row():
        with gr.Column():
            prompt = gr.Textbox(label="Texto (Prompt)", placeholder="Descreva a imagem que deseja gerar...")
            submit_button = gr.Button("Gerar Imagem")
        
        with gr.Column():
            output_image = gr.Image(label="Imagem Gerada")

    submit_button.click(fn=generate_image, inputs=prompt, outputs=output_image)

# Executar o app
if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860)