Update handler.py

handler.py

@@ -1,3 +1,12 @@
+from typing import Dict, List, Any
+from PIL import Image
+import torch
+from io import BytesIO
+from transformers import BlipForConditionalGeneration, BlipProcessor
+
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
         self.processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")