Update descriptions and default caption

README.md

@@ -1,5 +1,5 @@
 ---
-title: CapRL-Qwen3VL-4B
+title: CapRL-Qwen3VL-4B for Image Captioning
 emoji: 🎨
 colorFrom: blue
 colorTo: purple
@@ -8,6 +8,7 @@ sdk_version: 5.49.1
 app_file: app.py
 pinned: false
 license: apache-2.0
+short_description: Generate captions for images with CapRL-Qwen3VL-4B
 ---
 
 # CapRL-Qwen3VL-4B Image Captioning

app.py

@@ -10,77 +10,61 @@ MAX_NEW_TOKENS = 4096
 
 # Default demo content
 DEFAULT_IMAGE = "./examples/1909.png"
-DEFAULT_CAPTION = (
-    "The image is a bar chart from the Pew Research Center that illustrates how older Republicans and Republican leaners view Donald Trump, specifically focusing on how many describe the phrase \"fights for what I believe in\" to describe Trump. The data is based on a survey conducted from February 4-15, 2020, among U.S. adults who identify as Republicans or Republican-leaning independents.\n\n"
-    "### Title:\n"
-    "Older Republicans especially likely to see Trump as fighting for their beliefs\n\n"
-    "### Main Question:\n"
-    "Among Republicans and Republican leaners, % who say the phrase 'fights for what I believe in' describes Trump ...\n\n"
-    "### Data Breakdown:\n\n"
-    "1. **All Rep/Lean Rep (Overall):**\n"
-    "   - Very well: 51%\n"
-    "   - Fairly well: 36%\n"
-    "   - NET: 87%\n\n"
-    "2. **Ages 18-29:**\n"
-    "   - Very well: 31%\n"
-    "   - Fairly well: 45%\n"
-    "   - NET: 76%\n\n"
-    "3. **30-49:**\n"
-    "   - Very well: 41%\n"
-    "   - Fairly well: 42%\n"
-    "   - NET: 82%\n\n"
-    "4. **50-64:**\n"
-    "   - Very well: 58%\n"
-    "   - Fairly well: 33%\n"
-    "   - NET: 92%\n\n"
-    "5. **65+:**\n"
-    "   - Very well: 68%\n"
-    "   - Fairly well: 26%\n"
-    "   - NET: 94%\n\n"
-    "6. **Postgrad:**\n"
-    "   - Very well: 42%\n"
-    "   - Fairly well: 38%\n"
-    "   - NET: 80%\n\n"
-    "7. **College grad:**\n"
-    "   - Very well: 45%\n"
-    "   - Fairly well: 40%\n"
-    "   - NET: 85%\n\n"
-    "8. **Some college:**\n"
-    "   - Very well: 51%\n"
-    "   - Fairly well: 36%\n"
-    "   - NET: 87%\n\n"
-    "9. **HS or less:**\n"
-    "   - Very well: 56%\n"
-    "   - Fairly well: 33%\n"
-    "   - NET: 89\n\n"
-    "10. **Conserv (Conservative):**\n"
-    "    - Very well: 63%\n"
-    "    - Fairly well: 31%\n"
-    "    - NET: 94%\n\n"
-    "11. **Mod/Lib (Moderate/Liberal):**\n"
-    "    - Very well: 32%\n"
-    "    - Fairly well: 44%\n"
-    "    - NET: 75\n\n"
-    "12. **Republican:**\n"
-    "    - Very well: 61%\n"
-    "    - Fairly well: 32%\n"
-    "    - NET: 93\n\n"
-    "13. **Lean Republican:**\n"
-    "    - Very well: 36%\n"
-    "    - Fairly well: 41%\n"
-    "    - NET: 77\n\n"
-    "### Notes:\n"
-    "- The note at the bottom states that the data is based on Republicans and Republican-leaning independents.\n"
-    "- The source is a survey of U.S. adults conducted from February 4-15, 2020.\n\n"
-    "### Key Observations:\n"
-    "1. Older Republicans (65+) are the most likely to see Trump as someone who \"fights for what I believe in,\" with a net positive percentage of 94.\n"
-    "2. Younger age groups (18-29) have the lowest net positive percentage at 76.\n"
-    "3. Those with higher educational backgrounds (postgrad and college grad) have slightly lower net positive percentages compared to those with some college education (80 vs. 85).\n"
-    "4. Conservatives (63% very well) are the most likely to see Trump this way, followed by Republicans (61%).\n"
-    "5. Lean Republicans (36% very well) have the lowest percentage among the leaner categories.\n\n"
-    "This detailed description should provide a pure text model with sufficient information to answer any related questions about the image."
-)
-DEFAULT_CAPTION_TOKENS = 826
+DEFAULT_CAPTION = """Based on the provided bar chart from the Pew Research Center, here is a detailed description:
+
+**Title:** Older Republicans especially likely to see Trump as fighting for their beliefs
+**Subtitle:** Among Republicans and Republican leaners, % who say the phrase 'fights for what I believe in' describes Trump ...
+
+**Source:** Survey of U.S. adults conducted Feb. 4-15, 2020.
+
+**Legend:**
+-   **Very well:** 51% (All Rep/Lean Rep)
+-   **Fairly well:** 36% (All Rep/Lean Rep)
+-   **NET:** Sum of "Very well" and "Fairly well"
+
+**Data Summary:**
+
+**All Rep/Lean Rep:**
+-   Very well: 51%
+-   Fairly well: 36%
+-   NET: 87%
+
+**By Age:**
+-   Ages 18-29: Very well 31%, Fairly well 45%, NET 76%
+-   30-49: Very well 41%, Fairly well 42%, NET 82%
+-   50-64: Very well 58%, Fairly well 33%, NET 92%
+-   65+: Very well 68%, Fairly well 26%, NET 94%
+
+**By Education:**
+-   Postgrad: Very well 42%, Fairly well 38%, NET 80%
+-   College grad: Very well 45%, Fairly well 40%, NET 85%
+-   Some college: Very well 51%, Fairly well 36%, NET 87%
+-   HS or less: Very well 56%, Fairly well 33%, NET 89%
+
+**By Conservatism:**
+-   Conserv: Very well 63%, Fairly well 31%, NET 94%
+-   Mod/Lib: Very well 32%, Fairly well 44%, NET 75%
+
+**By Party Identification:**
+-   Republican: Very well 61%, Fairly well 32%, NET 93%
+-   Lean Republican: Very well 36%, Fairly well 41%, NET 77%
+
+**Analysis:**
+
+-   The overall percentage of Republicans and Republican leaners who say Trump "fights for what I believe in" is 87% (51% "very well" and 36% "fairly well").
+-   The group most likely to say this is those aged 65 and older (94% NET), followed by those 50-64 (92% NET) and those with a high school diploma or less (89% NET).
+-   The youngest group (18-29) is the least likely (76% NET).
+-   Among education levels, those with a high school diploma or less are most likely (89% NET), followed by some college (87%) and college graduates (85%).
+-   The most conservative group (Conservative) is the most likely to say this (94% NET), while the moderate/liberal group is the least likely (75% NET).
+-   Among party identifiers, Republicans are most likely (93% NET), while lean Republicans are less likely (77% NET).
+-   The 65+ age group is also the most conservative (94% NET) and the most likely to say Trump fights for their beliefs.
+-   The 65+ group is also the most likely to say it "very well" (68%) and the least likely to say it "fairly well" (26%).
+-   The most conservative group (63% "very well") is also the most likely overall (94% NET), while the most moderate/liberal group (32% "very well") is the least likely (75% NET).
+-   The Republican party has a 93% NET, while the lean Republican group has a 77% NET.
+
+**Conclusion:**
+The chart shows that older Republicans (65+) are the most likely to see Trump as fighting for their beliefs, with 94% saying it "very well" or "fairly well." This is followed by those aged 50-64 (92%) and those with a high school diploma or less (89%). The most conservative Republicans are also the most likely (94%), while moderate/liberal Republicans are the least likely (75%). The youngest group (18-29) is the least likely (76%)."""
+DEFAULT_CAPTION_TOKENS = 993
 
 
 def get_device() -> str:
@@ -181,15 +165,31 @@ def generate_caption(image: Image.Image):
 
 
 with gr.Blocks(title="CapRL-Qwen3VL-4B Image Captioning") as demo:
-    gr.Markdown("# CapRL-Qwen3VL-4B for Image Captioning")
-    gr.Markdown("### CapRL: Stimulating Dense Image Caption Capabilities via Reinforcement Learning")
-    gr.Markdown("Upload an image to generate a detailed caption with CapRL-Qwen3VL-4B!")
     gr.Markdown(
         """
-<a href="https://arxiv.org/abs/2509.22647">Paper</a> | <a href="https://github.com/InternLM/CapRL">Github</a> | <a href="https://huggingface.co/internlm/CapRL-Qwen3VL-4B">CapRL-Qwen3VL-4B Model</a> | <a href="https://huggingface.co/internlm/CapRL-3B">CapRL-3B Model</a> |
-<a href="https://huggingface.co/datasets/internlm/CapRL-2M">CapRL-2M Dataset</a>
+# CapRL
+📖<a href="https://arxiv.org/abs/2509.22647">Paper</a> | 🏠<a href="https://github.com/InternLM/CapRL">Github</a> | 🤗<a href="https://huggingface.co/collections/long-xing1/caprl-68d64ac32ded31596c36e189">CapRL Collection</a> | 🤗<a href="https://huggingface.co/papers/2509.22647">Daily Paper</a>
+
+### CapRL Series Model & Dataset
+| Series | Models & Resources |
+| :--- | :--- |
+| **CapRL 2.0 Series** | [🤗 CapRL-Qwen3VL-2B](https://huggingface.co/internlm/CapRL-Qwen3VL-2B) \| [🤗 CapRL-Qwen3VL-4B](https://huggingface.co/internlm/CapRL-Qwen3VL-4B) |
+| **CapRL 1.0 Series** | [🤗 CapRL-Qwen2.5VL-3B](https://huggingface.co/internlm/CapRL-3B) \| [🤗 CapRL-InternVL3.5-8B](https://huggingface.co/yuhangzang/CapRL-InternVL3.5-8B) \| [📊 CapRL-2M Dataset](https://huggingface.co/datasets/internlm/CapRL-2M) \| [📦 CapRL-3B-GGUF](https://huggingface.co/mradermacher/CapRL-3B-GGUF) \| [📦 CapRL-3B-i1-GGUF](https://huggingface.co/mradermacher/CapRL-3B-i1-GGUF) |
+
+### CapRL-Qwen3VL-4B
+We are excited to release the **CapRL 2.0 series**: **CapRL-Qwen3VL-2B** and **CapRL-Qwen3VL-4B**. These models feature fewer parameters while delivering even more powerful captioning performance.
+Notably, **CapRL-Qwen3VL-2B outperforms both CapRL-Qwen2.5VL-3B and Qwen2.5VL-72B in captioning tasks**.
+This leap in efficiency is driven by our upgraded training recipe, which includes a more rigorous QA data filter and a significantly more diverse image dataset. We welcome everyone to try them out!
+
+
+When selecting between the available CapRL models, it's essential to consider the trade-off between performance and computational cost.
+This guide will help you choose the most suitable model for your specific needs:
+|Model|Parameters|Strength|
+|-|-|-|
+|🤗[CapRL-Qwen3VL-2B](https://huggingface.co/internlm/CapRL-Qwen3VL-2B)|2B|Speed, Efficiency|
+|🤗[CapRL-Qwen3VL-4B](https://huggingface.co/internlm/CapRL-Qwen3VL-4B)|4B|High Performance, Advanced Captioning Ability|
 
-<a href="https://huggingface.co/collections/long-xing1/caprl-68d64ac32ded31596c36e189">CapRL Collection</a> | <a href="https://huggingface.co/papers/2509.22647">Daily Paper</a>
+**This Space** is based on **CapRL-Qwen3VL-4B**. You can also try out **CapRL-Qwen2.5VL-3B** 🎨&nbsp;&nbsp;&nbsp;&nbsp;➡️&nbsp;&nbsp;&nbsp;&nbsp;[🌈CapRL-Qwen2.5VL-3B Space](https://huggingface.co/spaces/yuhangzang/caprl)
 """
     )