Add pipeline tag and library name (#3)

- Add pipeline tag and library name (0d8656a623cd99ffb665bfa2e3e69c930a36fe47)


Co-authored-by: Niels Rogge <[email protected]>

README.md

@@ -1,7 +1,9 @@
 ---
 datasets:
 - yaolily/GenS-Video-150K
-license: other # Research use only
+license: other
+pipeline_tag: video-text-to-text
+library_name: transformers
 ---
 
 <p align="center">
@@ -149,7 +151,8 @@ def gens_frame_sampler(question: str, frame_paths: List[str], model, tokenizer,
     
     # Format the input as expected by the model
     frm_placeholders = ["<image1>" for _ in range(len(frames))]
-    content = "{}Question: {}\n{}".format("".join(frm_placeholders), question, prompt)
+    content = "{}Question: {}
+{}".format("".join(frm_placeholders), question, prompt)
     question_data = [{"role": "user", "content": content}]
     
     # Apply chat template
@@ -175,7 +178,8 @@ def gens_frame_sampler(question: str, frame_paths: List[str], model, tokenizer,
     
     # Decode and extract the relevant part of the response
     response = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
-    result = response.split("assistant\n")[-1].split("<|im_end|>")[0].strip()
+    result = response.split("assistant
+")[-1].split("<|im_end|>")[0].strip()
     
     return result