Update app.py

app.py

@@ -58,7 +58,6 @@ try:
                 tuned_unet_repo_id,
                 torch_dtype=torch_dtype, # Carrega com mesmo dtype
                 token=hf_token # Usa o token para acessar repo privado
-                # low_cpu_mem_usage=False # Pode precisar desativar se der erro OOM aqui
             )
             print("UNet treinado carregado. Substituindo UNet no pipeline...")
             pipe.unet = tuned_unet # A SUBSTITUIÇÃO
@@ -74,7 +73,6 @@ try:
          print(loading_error_message)
          DESCRIPTION += "\n" + loading_error_message
 
-
     # Configura o scheduler (como antes)
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     print(f"Scheduler configurado para: {pipe.scheduler.__class__.__name__}")
@@ -90,7 +88,6 @@ except Exception as e:
     DESCRIPTION += "\n" + loading_error_message
     pipe = None
 
-
 # Função generate (sem alterações significativas)
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed: seed = random.randint(0, MAX_SEED)
@@ -100,7 +97,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 def generate(
     prompt: str,
     negative_prompt: str = "",
-    use_negative_prompt: bool = True,
+    use_negative_prompt: bool = False,
     seed: int = 0,
     width: int = 1024,
     height: int = 1024,
@@ -125,14 +122,19 @@ def generate(
         print(f"Erro durante a geração: {e}")
         raise gr.Error(f"Erro durante a geração: {e}")
 
-
-# Interface Gradio (sem alterações significativas)
-examples = [ "photo of a futuristic city skyline at sunset, high detail", "an oil painting of a cute cat wearing a wizard hat", "Astronaut in a jungle, cold color palette, oil pastel, detailed, 8k", "An alien grasping a sign board contain word 'Flash', futuristic, neonpunk, detailed" ]
+# Interface Gradio
+examples = [
+    "photo of a futuristic city skyline at sunset, high detail",
+    "an oil painting of a cute cat wearing a wizard hat",
+    "Astronaut in a jungle, cold color palette, oil pastel, detailed, 8k",
+    "An alien grasping a sign board contain word 'Flash', futuristic, neonpunk, detailed"
+]
 css = ".gradio-container { max-width: 800px !important; margin: 0 auto !important; } h1{ text-align:center }"
 with gr.Blocks(css=css) as demo:
     gr.Markdown(f"""# SDXL Base com UNet Fine-tuned (`{tuned_unet_repo_id}`)
         Base: `{base_model_id}`
-        {DESCRIPTION}""")
+        {DESCRIPTION}
+        **Aviso:** O filtro de conteúdo explícito foi desativado. Use prompts com cuidado.""")
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(label="Prompt", show_label=False, max_lines=3, placeholder="Descreva a imagem...", container=False)
@@ -140,8 +142,15 @@ with gr.Blocks(css=css) as demo:
         result = gr.Gallery(label="Resultado", show_label=False, elem_id="gallery", columns=1, height=768)
     with gr.Accordion("Opções Avançadas", open=False):
         with gr.Row():
-            use_negative_prompt = gr.Checkbox(label="Usar prompt negativo", value=True)
-            negative_prompt = gr.Text(label="Prompt Negativo", max_lines=3, lines=2, placeholder="O que NÃO ver...", value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW", visible=True)
+            use_negative_prompt = gr.Checkbox(label="Usar prompt negativo", value=False)
+            negative_prompt = gr.Text(
+                label="Prompt Negativo",
+                max_lines=3,
+                lines=2,
+                placeholder="O que evitar na imagem (ex.: blurry, deformed)...",
+                value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, blurry, amputation",
+                visible=False
+            )
         seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
         randomize_seed = gr.Checkbox(label="Seed Aleatória", value=True)
         with gr.Row():