Update app.py
Browse files
app.py
CHANGED
|
@@ -395,8 +395,8 @@ caption_model.fit(
|
|
| 395 |
)
|
| 396 |
|
| 397 |
#Carga del modelo
|
| 398 |
-
|
| 399 |
-
caption_model.
|
| 400 |
|
| 401 |
#Interfaz para gradio
|
| 402 |
def generate_caption(sample_img):
|
|
|
|
| 395 |
)
|
| 396 |
|
| 397 |
#Carga del modelo
|
| 398 |
+
archivo_pesos = os.path.join("pesos10.npy")
|
| 399 |
+
caption_model = np.load(archivo_pesos, allow_pickle=True)
|
| 400 |
|
| 401 |
#Interfaz para gradio
|
| 402 |
def generate_caption(sample_img):
|