Update handler.py

handler.py

@@ -5,7 +5,7 @@ from transformers import AutoModelForCausalLM, AutoProcessor
 from transformers.image_utils import to_numpy_array, PILImageResampling, ChannelDimension
 from transformers.image_transforms import resize, to_channel_dimension_format
 
-class CustomPipeline:
+class EndpointHandler:
     def __init__(self, model_path: str, api_token: str):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.processor = AutoProcessor.from_pretrained(