fix import for the model-5

handler.py

@@ -1,10 +1,3 @@
-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 LlavaNextVideoForConditionalGeneration, LlavaNextVideoProcessor
@@ -12,8 +5,8 @@ from transformers import LlavaNextVideoForConditionalGeneration, LlavaNextVideoP
 class EndpointHandler:
     def __init__(self, path=""):
         # Load the model and processor
-        self.model = LlavaNextVideoForConditionalGeneration.from_pretrained(path)
-        self.processor = LlavaNextVideoProcessor.from_pretrained(path)
+        self.model = LlavaNextVideoForConditionalGeneration.from_pretrained('EnariGmbH/surftown-1.0')
+        self.processor = LlavaNextVideoProcessor.from_pretrained('EnariGmbH/surftown-1.0')
 
         # Ensure the model is in evaluation mode
         self.model.eval()