Update app.py

app.py

@@ -11,8 +11,13 @@ model_repo_id = "stabilityai/sdxl-turbo"  # Continuando com o modelo especificad
 
 # Carregar o pipeline configurado para CPU
 pipe = DiffusionPipeline.from_pretrained(model_repo_id)
+revision="fp16",  # Adicione esta linha para otimização, se aplicável
+)
 pipe = pipe.to(device)
 
+# Atualize as configurações do pipeline diretamente
+pipe.unet.config.sample_size = (808, 512)  # Ajuste de altura e largura divisíveis por 8
+
 # Carregando o dataset do Hugging Face
 dataset = load_dataset("LEIDIA/Data_Womleimg")
 
@@ -67,6 +72,13 @@ def resize_to_divisible_by_8(image):
     new_height = height + (8 - height % 8) if height % 8 != 0 else height
     return image.resize((new_width, new_height))
 
+def infer_simple(prompt):
+    # Garantir que as dimensões sejam divisíveis por 8 antes de gerar
+    adjusted_height = IMAGE_HEIGHT - (IMAGE_HEIGHT % 8)
+    adjusted_width = IMAGE_WIDTH - (IMAGE_WIDTH % 8)
+
+    print(f"Using height: {IMAGE_HEIGHT} and width: {IMAGE_WIDTH}")
+
 # Função simples para gerar imagem
 def infer_simple(prompt):
     # Geração da imagem