Update model.py

model.py

@@ -308,7 +308,7 @@ def load_image_from_path(img_path):
     return img
 
 
-def generate_caption(img_path, add_noise=False):
+def generate_caption(img_path,caption_model, add_noise=False):
     img = load_image_from_path(img_path)
     
     if add_noise: