Update handler.py

handler.py

@@ -15,9 +15,7 @@ class EndpointHandler():
         # load the optimized model
         
         self.processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base") 
-        self.model = BlipForConditionalGeneration.from_pretrained(
-            "Salesforce/blip-image-captioning-base"
-        ).to(device)
+        self.model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to(device)
         self.model.eval()
         self.model = self.model.to(device)
         
@@ -45,9 +43,7 @@ class EndpointHandler():
         processed_images = {**processed_images, **parameters}
         
         with torch.no_grad():
-            out = self.model.generate(
-                **processed_images
-            )
+            out = self.model.generate(**processed_images)
         captions = self.processor.batch_decode(out, skip_special_tokens=True)
 
         return {"captions": captions}