fix #7

app.py

@@ -8,18 +8,18 @@ autoencoder = load_model("models/autoencoder.h5", custom_objects={"mse": MeanSqu
 decoder = load_model("models/decoder.h5")
 
 latent_dim = 128 
-import tensorflow as tf
 
 
 def add_gaussian_noise(image, noise_factor=0.2):
     noisy_image = image + noise_factor * np.random.normal(size=image.shape)
     noisy_image = np.clip(noisy_image, 0., 1.)
-    noisy_image = tf.image.resize(noisy_image, (224, 224))
     return noisy_image
 
 def denoise_and_generate(image, num_images):
     image = np.array(image) / 255.0
+    image = tf.image.resize(image, (224, 224))  # Redimensionar la imagen
     noisy_image = add_gaussian_noise(np.expand_dims(image, axis=0))
+    
     denoised_image = autoencoder.predict(noisy_image).squeeze()
     denoised_image = (denoised_image * 255).astype(np.uint8)
     
@@ -32,6 +32,7 @@ def denoise_and_generate(image, num_images):
     
     return outputs
 
+
 # Define la interfaz
 inputs = [
     gr.Image(label="Imagen de Entrada"),