Update handler.py

handler.py

@@ -8,8 +8,8 @@ class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
         self.device = "cuda:0"
-        self.model = AutoModelForVision2Seq.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf").to(self.device)
-        self.processor = AutoProcessor.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf")
+        self.model = AutoModelForVision2Seq.from_pretrained("HuggingFaceM4/idefics2-8b").to(self.device)
+        self.processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b")
 
     def __call__(self, data: dict[str, Any]) -> str:
         """