fix import for the model-4

handler.py

@@ -1,12 +1,19 @@
+import transformers
+import logging
+
+# Print the transformers package version
+logging.basicConfig(level=logging.INFO)
+logging.info(f"Transformers version: {transformers.__version__}")
+
 from typing import Dict, List, Any
 import torch
-from transformers import VideoLlavaProcessor, VideoLlavaForConditionalGeneration
+from transformers import LlavaNextVideoForConditionalGeneration, LlavaNextVideoProcessor
 
 class EndpointHandler:
     def __init__(self, path=""):
         # Load the model and processor
-        self.model = VideoLlavaForConditionalGeneration.from_pretrained(path)
-        self.processor = VideoLlavaProcessor.from_pretrained(path)
+        self.model = LlavaNextVideoForConditionalGeneration.from_pretrained(path)
+        self.processor = LlavaNextVideoProcessor.from_pretrained(path)
 
         # Ensure the model is in evaluation mode
         self.model.eval()