EnariGmbH
/

surftown-1.0

llava_next_video

Inference Endpoints

Model card Files Files and versions Community

VascoBartolo commited on Sep 3, 2024

Commit

05d1a5d

·

1 Parent(s): 16d7281

fix import for the model-6

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from transformers import LlavaNextVideoForConditionalGeneration, LlavaNextVideoProcessor
 class EndpointHandler:
-    def __init__(self, path=""):
         # Load the model and processor
         self.model = LlavaNextVideoForConditionalGeneration.from_pretrained('EnariGmbH/surftown-1.0')
         self.processor = LlavaNextVideoProcessor.from_pretrained('EnariGmbH/surftown-1.0')

 from transformers import LlavaNextVideoForConditionalGeneration, LlavaNextVideoProcessor
 class EndpointHandler:
+    def __init__(self):
         # Load the model and processor
         self.model = LlavaNextVideoForConditionalGeneration.from_pretrained('EnariGmbH/surftown-1.0')
         self.processor = LlavaNextVideoProcessor.from_pretrained('EnariGmbH/surftown-1.0')