Update handler.py

handler.py

@@ -8,8 +8,8 @@ from io import BytesIO
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-6.7b")
-        self.model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-6.7b")
+        self.processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
+        self.model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b")
 
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.model.to(self.device)