Update handler.py

handler.py

@@ -3,16 +3,15 @@ from PIL import Image
 import torch
 import base64
 from io import BytesIO
-from transformers import BlipProcessor, BlipForQuestionAnswering
+from transformers import AutoProcessor, BlipForQuestionAnswering
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-capfilt-large")
+        self.processor = AutoProcessor.from_pretrained("Salesforce/blip-vqa-capfilt-large")
         self.model = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-capfilt-large").to(device)
 
-
     def __call__(self, data: Any) -> List[float]:
         inputs = data.pop("inputs", data)