ojoel98 commited on
Commit
8f47587
·
verified ·
1 Parent(s): b05b888

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -395,8 +395,8 @@ caption_model.fit(
395
  )
396
 
397
  #Carga del modelo
398
- pesos_nuevos = np.load('https://huggingface.co/spaces/ojoel98/App2/blob/main/pesos10.npy', allow_pickle=True)
399
- caption_model.set_weights(pesos_nuevos)
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):