app.py

import gradio as gr
from PIL import Image
import torch
from torchvision import transforms
from transformers import pipeline

# Carregar o pipeline Hugging Face para colorização
colorization_pipeline = pipeline("image-colorization", model="huggingface/ColorfulImageColorization")

# Função para colorizar a imagem
def colorize_image(image):
    image = image.convert("RGB")
    colorized = colorization_pipeline(image)
    return colorized

# Configuração da Interface Gradio
interface = gr.Interface(
    fn=colorize_image,
    inputs=gr.Image(type="pil", label="Upload de Imagem em Preto e Branco"),
    outputs=gr.Image(type="pil", label="Imagem Colorida"),
    title="Colorização de Imagens com IA",
    description="Faça upload de uma imagem em preto e branco, e esta IA a colorizará automaticamente!"
)

# Executar a aplicação
interface.launch()