Update app.py

app.py

@@ -30,7 +30,7 @@ def image_preprocess(image):
         image = image.convert("RGB")
     return image
 
-def generate_caption(image_processor, model, device, image):
+def generate_caption(image_processor, model, tokenizer,device, image):
     inputs = image_processor(image, return_tensors='pt').to(device)
     model.eval()
     # Generate caption
@@ -64,12 +64,11 @@ def main():
             if st.sidebar.button('Generate Caption'):
                 with st.spinner('Generating caption...'):
                     image_processor, model,tokenizer, device = initialize_model()
-                    caption = generate_caption(image_processor, model, device, image)
+                    caption = generate_caption(image_processor, model, tokenizer,device, image)
             
                     st.header("Caption:")
                     st.markdown(f'**{caption}**')
 
-
 if __name__ == '__main__':
     main()