Update app.py

app.py

@@ -40,7 +40,7 @@ dataset_descriptions = [
 ]
 
 # Definir parâmetros padrão para geração rápida
-DEFAULT_PROMPT = "A beautiful brunette woman wearing a leather outfit"
+DEFAULT_PROMPT = "A beautiful brunette woman wearing a blue leather pants "
 DEFAULT_INFERENCE_STEPS = 6
 IMAGE_WIDTH = 512
 IMAGE_HEIGHT = 812
@@ -54,18 +54,22 @@ def resize_to_divisible_by_8(image):
 
 # Função simples para gerar imagem
 def infer_simple(prompt):
+    # Ajustando as dimensões para serem divisíveis por 8
+    adjusted_width = IMAGE_WIDTH + (8 - IMAGE_WIDTH % 8) if IMAGE_WIDTH % 8 != 0 else IMAGE_WIDTH
+    adjusted_height = IMAGE_HEIGHT + (8 - IMAGE_HEIGHT % 8) if IMAGE_HEIGHT % 8 != 0 else IMAGE_HEIGHT
+
     # Geração da imagem
     image = pipe(
         prompt=prompt,
         num_inference_steps=DEFAULT_INFERENCE_STEPS,
         guidance_scale=GUIDANCE_SCALE,
-        height=IMAGE_HEIGHT,
-        width=IMAGE_WIDTH,
+        height=adjusted_height,
+        width=adjusted_width,
     ).images[0]
-    # Redimensionar a imagem para ser divisível por 8
-    image = resize_to_divisible_by_8(image)
+    
     return image
 
+
 # Interface Gradio
 with gr.Blocks() as demo:
     with gr.Row():