Update app.py

app.py

@@ -1,9 +1,8 @@
 import gradio as gr
-import numpy as np
-import random
 import torch
 from diffusers import DiffusionPipeline
 from datasets import load_dataset
+from PIL import Image
 
 
 # Configurações do dispositivo para uso apenas da CPU
@@ -53,11 +52,9 @@ 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))
 
-image = pipe(prompt, width=width, height=height)
-image = resize_to_divisible_by_8(image)
-
 # Função simples para gerar imagem
 def infer_simple(prompt):
+    # Geração da imagem
     image = pipe(
         prompt=prompt,
         num_inference_steps=DEFAULT_INFERENCE_STEPS,
@@ -65,6 +62,8 @@ def infer_simple(prompt):
         height=IMAGE_HEIGHT,
         width=IMAGE_WIDTH,
     ).images[0]
+    # Redimensionar a imagem para ser divisível por 8
+    image = resize_to_divisible_by_8(image)
     return image
 
 # Interface Gradio