Upload folder using huggingface_hub

app.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2024 Ye Liu. Licensed under the BSD-3-Clause license.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause license.
 
 import html
 import json
@@ -28,7 +28,7 @@ MODEL_HF = 'yeliudev/VideoMind-2B'
 
 TITLE = 'VideoMind: A Chain-of-LoRA Agent for Long Video Reasoning'
 
-TITLE_MD = f'<h1 align="center">💡 {TITLE}</h1>'
+LOGO_MD = '<p align="center"><img width="350" src="https://raw.githubusercontent.com/yeliudev/VideoMind/refs/heads/main/.github/logo.png"></p>'
 DESCRIPTION_MD = """VideoMind is a multi-modal agent framework that enhances video reasoning by emulating *human-like* processes, such as *breaking down tasks*, *localizing and verifying moments*, and *synthesizing answers*. This approach addresses the unique challenges of temporal-grounded reasoning in a progressive strategy. Please find more details at our <a href="https://videomind.github.io/" target="_blank">Project Page</a>, <a href="https://arxiv.org/abs/2503.13444" target="_blank">Tech Report</a> and <a href="https://github.com/yeliudev/VideoMind" target="_blank">GitHub Repo</a>."""  # noqa
 
 # yapf:disable
@@ -580,7 +580,7 @@ if __name__ == '__main__':
     prompt = gr.Textbox(label='Text Prompt', placeholder='Ask a question about the video...')
 
     with gr.Blocks(title=TITLE, css=CSS, js=JS) as demo:
-        gr.Markdown(TITLE_MD)
+        gr.Markdown(LOGO_MD)
         gr.Markdown(DESCRIPTION_MD)
 
         with gr.Row():
@@ -633,4 +633,4 @@ if __name__ == '__main__':
             with gr.Column(scale=5):
                 chat.render()
 
-        demo.launch()
+        demo.launch(server_name='0.0.0.0')

videomind/constants.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 IGNORE_INDEX = -100
 

videomind/conversation.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 from dataclasses import dataclass
 from typing import List

videomind/dataset/collator.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import warnings
 

videomind/dataset/hybrid.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import math
 import random

videomind/dataset/sub_classes/activitynet_captions.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 from collections import OrderedDict
 

videomind/dataset/sub_classes/activitynet_rtl.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import re
 from collections import OrderedDict

videomind/dataset/sub_classes/cgbench.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 from torch.utils.data import Dataset

videomind/dataset/sub_classes/charades_sta.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/cosmo_cap.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/didemo.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import random
 

videomind/dataset/sub_classes/ego4d_naq.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 from collections import OrderedDict
 

videomind/dataset/sub_classes/ego4d_nlq.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/ego_timeqa.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import random
 

videomind/dataset/sub_classes/hirest.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 from collections import OrderedDict
 

videomind/dataset/sub_classes/internvit_vtime.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/longvideobench.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 from torch.utils.data import Dataset

videomind/dataset/sub_classes/lvbench.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 from torch.utils.data import Dataset

videomind/dataset/sub_classes/mlvu.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 from torch.utils.data import Dataset

videomind/dataset/sub_classes/mvbench.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 from torch.utils.data import Dataset

videomind/dataset/sub_classes/nextgqa.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import csv
 

videomind/dataset/sub_classes/nextqa.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import csv
 

videomind/dataset/sub_classes/qa_ego4d.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import random
 

videomind/dataset/sub_classes/queryd.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/qvhighlights.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/rextime.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/star.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/tacos.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/vid_morp.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/videomme.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 from torch.utils.data import Dataset

videomind/dataset/sub_classes/videoxum.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import nncore
 

videomind/dataset/sub_classes/youcook2.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 from collections import OrderedDict
 

videomind/dataset/utils.py

@@ -202,12 +202,9 @@ def _read_video_decord(ele: dict, ) -> torch.Tensor:
     # TODO: the actual total_frames shall be computed by e_frame - s_frame + 1
     # but it would affect verifier's performance when video_start and video_end get clamped
     # shall be fixed by using normalized timestamps instead of real time
-    total_frames = min(max(1, round((e - s) * video_fps)), total_frames)
+    total_frames = min(max(FPS_MIN_FRAMES, round((e - s) * video_fps)), total_frames)
 
-    if total_frames > FPS_MIN_FRAMES:
-        nframes = smart_nframes(ele, total_frames=total_frames, video_fps=video_fps)
-    else:
-        nframes = total_frames
+    nframes = smart_nframes(ele, total_frames=total_frames, video_fps=video_fps)
 
     # 2. generate frame ids
     idx = torch.linspace(s_frame, e_frame, nframes).round().long().tolist()

videomind/dataset/wrappers/answering.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import copy
 import random

videomind/dataset/wrappers/grounding.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import copy
 

videomind/dataset/wrappers/planning.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import copy
 

videomind/dataset/wrappers/verifying.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import copy
 

videomind/eval/eval_auto.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import argparse
 

videomind/eval/infer_auto.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import argparse
 import copy

videomind/eval/infer_qvhighlights.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import argparse
 import copy

videomind/model/blocks.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import math
 

videomind/model/builder.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import warnings
 

videomind/model/generator.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import torch
 import torch.nn as nn

videomind/model/loss.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import math
 import warnings

videomind/model/model.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import warnings
 

videomind/train/custom_trainer.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import warnings
 

videomind/train/train.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 from dataclasses import dataclass, field
 from typing import Optional

videomind/utils/io.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import pysrt
 from decord import VideoReader

videomind/utils/parser.py

@@ -1,4 +1,4 @@
-# Copyright (c) Ye Liu. Licensed under the BSD 3-Clause License.
+# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
 
 import re