Update app.py

app.py

@@ -203,8 +203,8 @@ def generate_image(model_name: str, text: str, image: Image.Image,
         images=[image],
         return_tensors="pt",
         padding=True,
-        #truncation=True,
-        #max_length=MAX_INPUT_TOKEN_LENGTH
+        truncation=True,
+        max_length=MAX_INPUT_TOKEN_LENGTH
     ).to(device)
     streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
     generation_kwargs = {**inputs, "streamer": streamer, "max_new_tokens": max_new_tokens}
@@ -259,8 +259,8 @@ def generate_video(model_name: str, text: str, video_path: str,
         add_generation_prompt=True,
         return_dict=True,
         return_tensors="pt",
-        #truncation=True,
-        #max_length=MAX_INPUT_TOKEN_LENGTH
+        truncation=True,
+        max_length=MAX_INPUT_TOKEN_LENGTH
     ).to(device)
     streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
     generation_kwargs = {