Update app.py

app.py

@@ -1,11 +1,12 @@
 import streamlit as st
-from transformers import VisionEncoderDecoderModel, ViTImageProcessor, GPT2Tokenizer
+from transformers import VisionEncoderDecoderModel, ViTImageProcessor, GPT2Tokenizer, AutoImageProcessor
 import torch
 from PIL import Image
 
 # Load the model and tokenizer
 model = VisionEncoderDecoderModel.from_pretrained("ashok2216/vit-gpt2-image-captioning_COCO_FineTuned")
-processor = ViTImageProcessor.from_pretrained("ashok2216/vit-gpt2-image-captioning_COCO_FineTuned")
+# processor = ViTImageProcessor.from_pretrained("ashok2216/vit-gpt2-image-captioning_COCO_FineTuned")
+processor = AutoImageProcessor.from_pretrained("ashok2216/vit-gpt2-image-captioning_COCO_FineTuned")
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 
 # Streamlit app title