Commit Graph

5 Commits

Author SHA1 Message Date
JamePeng
b109c67478 fixed missing import 2024-08-27 20:52:52 +08:00
JamePeng
344ddc2cb1 Optimize video frame sampling logic
- Replaced manual index calculation with `np.linspace` for improved efficiency and readability.
- Reduced computation overhead by utilizing NumPy's vectorized operations for generating evenly spaced frame indices.
2024-08-27 04:40:48 +08:00
JamePeng
3745c3316a Update web_demo_streamlit-minicpmv2_6.py 2024-08-27 04:20:48 +08:00
JamePeng
dc5f809253 Update web_demo_streamlit-minicpmv2_6.py
1. Avoid using 'None' string when `user_text` is empty.
2. Added `st.spinner` to display a loading message during AI content generation.
2024-08-27 04:03:35 +08:00
JamePeng
c9f5cd4b00 Update streamlit implementation for MiniCPM-V 2.6
Compared with the streamlit implementation of 2.5, this code implementation can better play the new multi-modal capabilities of 2.6:
1. The application supports the upload and processing of text, single image, multiple images and videos, and can process different types of input according to the mode selected by the user.

2. Video frame extraction and encoding: In video mode, frames are extracted from the uploaded video through the decord library and uniformly sampled so that the model can process and generate responses. More detailed and clear variables and annotations. Convenient for learning and use

3. File upload and processing: Support users to upload pictures and videos, and perform corresponding processing according to different modes, such as displaying pictures in single picture mode, displaying multiple pictures in multi-picture mode, and processing video frames in video mode. You can switch back and forth between different media.

4. Tip: You can use the command `streamlit run ./web_demo_streamlit-minicpmv2_6.py --server.maxUploadSize 1024` 
to adjust the maximum upload size to 1024MB or larger files. The default 200MB limit of Streamlit's file_uploader component might be insufficient for video-based interactions. Adjust the size based on your GPU memory usage.
2024-08-19 00:45:33 +08:00