Update app.py

app.py

@@ -1,51 +1,43 @@
-# -*- coding: utf-8 -*-
-"""Untitled11.ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/17mNpy3A7oaJ-q9Txp8f4eBvRfO_k7NCb
-"""
-
-pip install diffusers
-pip install transformers
-
+import os
 import torch
 from diffusers import StableDiffusionImg2ImgPipeline
 from diffusers.utils import load_image
 import gradio as gr
 
-# Dispositivo de processamento
-device = "cuda"
+# Definir dispositivo de processamento
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Carregar o modelo CompVis/stable-diffusion-v1-4 (compatível com Img2ImgPipeline)
+# Carregar o modelo Stable Diffusion Img2Img
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
-    "CompVis/stable-diffusion-v1-4",  # Modelo compatível
+    "CompVis/stable-diffusion-v1-4",
     torch_dtype=torch.float16
 )
 pipe = pipe.to(device)
 
-# Carregar a imagem inicial
-url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg"
-init_image = load_image(url).resize((1024, 1024))
-
-# Prompt a ser usado para geração
-prompt = "cat wizard, gandalf, lord of the rings, detailed, fantasy, cute, adorable, Pixar, Disney, 8k"
-
-# Função de geração de imagem a partir de um prompt e uma imagem inicial
+# Função para gerar imagem
 def generate_image(prompt, image):
-    image = pipe(
+    # Processar a imagem com o modelo
+    result_image = pipe(
         prompt=prompt,
         image=image,
-        num_inference_steps=50,  # Aumentar o número de passos pode melhorar a qualidade
-        strength=0.75,  # Reduzir a força ajuda a manter mais da imagem original (rosto consistente)
-        guidance_scale=7.5  # Controle de qualidade para manter o rosto mais detalhado
+        num_inference_steps=50,
+        strength=0.75,
+        guidance_scale=7.5
     ).images[0]
-    return image
+    return result_image
 
-# Interface com Gradio para facilitar a geração de imagens
-gr.Interface(
+# Configurar a interface Gradio
+iface = gr.Interface(
     fn=generate_image,
-    inputs=["text", "image"],
-    outputs="image"
-).launch()
+    inputs=[
+        gr.Textbox(label="Prompt", placeholder="Digite o seu prompt aqui..."),
+        gr.Image(label="Imagem Inicial", type="pil")  # Espera uma imagem no formato PIL
+    ],
+    outputs=gr.Image(label="Imagem Gerada"),
+    title="Gerador de Imagens com Stable Diffusion Img2Img",
+    description="Insira um prompt e uma imagem inicial para gerar uma nova imagem."
+)
+
+# Executar a interface
+if __name__ == "__main__":
+    iface.launch()