Update app.py

app.py

@@ -554,18 +554,18 @@ caption_model.fit(
 """### **Opción para guardar el modelo entrenado**"""
 
 #con está opción vemos los pesos del modelo en una lista
-#pesos = caption_model.get_weights()
+pesos = caption_model.get_weights()
 
 #guardamos esos pesos en formato npy - en este caso lo guardamos entrenado con una época, ya que si quitamos el fit o el entrenamiento nos da error, por lo que siempre tenemos que
 #entrenarle al modelo con una época para después configurarle con otro con 10 épocas
-#np.save('pesos1.npy', np.array(pesos, dtype=object), allow_pickle=True)
+np.save('pesos1.npy', np.array(pesos, dtype=object), allow_pickle=True)
 
 #aquí configuramos los pesos que estaban entrenados con una época con diez - nosotros corrimos anteriormente con 10 y nos descargamos
 import os
 import numpy as np
 
 #Carga del modelo
-archivo_pesos = os.path.join("pesos10.npy")
+archivo_pesos = os.path.join("pesos1.npy")
 caption_model = np.load(archivo_pesos, allow_pickle=True)
 
 #Interfaz para gradio