chore: change model

handler.py

@@ -1,5 +1,5 @@
 # +
-from typing import  Dict, Any
+from typing import  Dict, List, Any
 from PIL import Image
 import torch
 import requests
@@ -13,10 +13,13 @@ class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
         
-        self.processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large") 
-        self.model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large").to(device)
+        self.processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base") 
+        self.model = BlipForConditionalGeneration.from_pretrained(
+            "Salesforce/blip-image-captioning-base"
+        ).to(device)
         self.model.eval()
         self.model = self.model.to(device)
+        
 
 
     def __call__(self, data: Any) -> Dict[str, Any]: