Update src/model.py

src/model.py

@@ -22,6 +22,24 @@ tokenizer = AutoTokenizer.from_pretrained(
 model.eval()
 
 
+class _GeneratorPickleHack:
+    def __init__(self, generator, generator_id=None):
+        self.generator = generator
+        self.generator_id = (
+            generator_id if generator_id is not None else str(uuid.uuid4())
+        )
+
+    def __call__(self, *args, **kwargs):
+        return self.generator(*args, **kwargs)
+
+    def __reduce__(self):
+        return (_GeneratorPickleHack_raise, (self.generator_id,))
+
+
+def _GeneratorPickleHack_raise(*args, **kwargs):
+    raise AssertionError("cannot actually unpickle _GeneratorPickleHack!")
+
+
 @spaces.GPU()
 def describe_video(video, question):
     """
@@ -35,7 +53,8 @@ def describe_video(video, question):
         str: The generated answer to the question.
     """
     # Encode the video frames
-    frames = encode_video(video)
+    frames = _GeneratorPickleHack(encode_video)(video)
+    #frames = encode_video(video)
     #frames = list(frames)  # Convert generator or any iterable to list
 
     # Message format for the model