replace NougatProcessor with AutoProcessor

config.json

@@ -184,5 +184,5 @@
   "model_type": "vision-encoder-decoder",
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
-  "transformers_version": "main"
+  "transformers_version": "4.34.0.dev0"
 }

handler.py

@@ -1,13 +1,13 @@
 from io import BytesIO
-from typing import Dict, List, Any
-from transformers import NougatProcessor, VisionEncoderDecoderModel
+from typing import Dict, Any
+from transformers import AutoProcessor, VisionEncoderDecoderModel
 from transformers.image_utils import base64
 from PIL import Image
 import torch
 
 class EndpointHandler():
     def __init__(self, path="facebook/nougat-base") -> None:
-        self.processor = NougatProcessor.from_pretrained(path)
+        self.processor = AutoProcessor.from_pretrained(path)
         self.model = VisionEncoderDecoderModel.from_pretrained(path)
 
         self.device = "cuda" if torch.cuda.is_available() else "cpu"