Create app.py

app.py

@@ -0,0 +1,26 @@
+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()