Update handler.py

handler.py

@@ -4,14 +4,15 @@ from PIL import Image
 import torch
 import base64
 from io import BytesIO
-from transformers import BlipForConditionalGeneration, BlipProcessor
+from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers.image_utils import load_image
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.processor = BlipProcessor.from_pretrained("zesquirrelnator/idefics2-8b-docvqa-finetuned-tutorial")
-        self.model = BlipForConditionalGeneration.from_pretrained(
+        self.processor = AutoProcessor.from_pretrained("zesquirrelnator/idefics2-8b-docvqa-finetuned-tutorial")
+        self.model = AutoModelForVision2Seq.from_pretrained(
             "zesquirrelnator/idefics2-8b-docvqa-finetuned-tutorial"
         ).to(device)
         self.model.eval()