fix #1

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 from tensorflow.keras.models import load_model
 
 # Carga los modelos previamente entrenados
-autoencoder = load_model("autoencoder.h5")
-decoder = load_model("decoder.h5")
+autoencoder = load_model("models/autoencoder.h5")
+decoder = load_model("models/decoder.h5")
 
 latent_dim = 128 
 
@@ -26,9 +26,6 @@ def denoise_and_generate(image, num_images):
     # Prepara las imágenes para devolverlas
     outputs = [denoised_image] + [generated_images[i].squeeze() for i in range(num_images)]
     
-    # Rellena los valores faltantes para que siempre haya 11 valores
-    outputs += [None] * (11 - len(outputs))
-    
     return outputs
 
 # Define la interfaz

requirements.txt

@@ -1,4 +1,4 @@
-tensorflow>=2.0.0
-gradio>=3.50.0
-numpy>=1.19.0
-pillow>=9.0.0
+tensorflow
+tensorflow-cpu
+gradio
+numpy