Update app.py
Browse files
app.py
CHANGED
|
@@ -565,8 +565,8 @@ import os
|
|
| 565 |
import numpy as np
|
| 566 |
|
| 567 |
#Carga del modelo
|
| 568 |
-
|
| 569 |
-
|
| 570 |
|
| 571 |
#Interfaz para gradio
|
| 572 |
def generate_caption(sample_img):
|
|
|
|
| 565 |
import numpy as np
|
| 566 |
|
| 567 |
#Carga del modelo
|
| 568 |
+
archivo_pesos = os.path.join("pesos10.npy")
|
| 569 |
+
caption_model = np.load(archivo_pesos, allow_pickle=True)
|
| 570 |
|
| 571 |
#Interfaz para gradio
|
| 572 |
def generate_caption(sample_img):
|