Update app.py

app.py

@@ -7,14 +7,16 @@ from huggingface_hub import hf_hub_download
 # Descargar modelo desde Hugging Face
 model_path = hf_hub_download(repo_id="Bmo411/DenoisingAutoencoder", filename="autoencoder_complete_model_Fourier.keras")
 
-# Cargar el modelo una sola vez para evitar recargas en cada ejecución
-model = tf.keras.models.load_model(model_path, compile=True)
+from tensorflow.keras.saving import register_keras_serializable
+
+@register_keras_serializable()
 def fourier_transform(x):
     fourier = tf.signal.fft2d(tf.cast(x, tf.complex64))
     fourier = tf.complex(tf.math.real(fourier), tf.math.imag(fourier))
     fourier = tf.abs(fourier)
-    return tf.concat([tf.math.real(fourier), tf.math.imag(fourier)], axis=-1)  # Separar parte real e imaginaria
+    return tf.concat([tf.math.real(fourier), tf.math.imag(fourier)], axis=-1)
 
+@register_keras_serializable()
 def inverse_fourier_transform(x):
     real_part, imag_part = tf.split(x, num_or_size_splits=2, axis=-1)
     complex_fourier = tf.complex(real_part, imag_part)