Improve model card: Add pipeline_tag, library_name, paper link, and sample usage

README.md

@@ -1,11 +1,13 @@
 ---
-license: apache-2.0
 language:
 - en
 - ko
+license: apache-2.0
+pipeline_tag: image-text-to-text
+library_name: transformers
 ---
 
-#  gWorld-8B 🌍📱 
+# gWorld-8B 🌍📱 
 
 <p align="center">
 <picture>
@@ -21,6 +23,7 @@ language:
 
 **gWorld-8B 🌍📱** is the first open-weight, single self-contained Vision-Language Model (VLM) specialized for visual mobile GUI world modeling. Unlike traditional visual world models that predict pixels directly, **gWorld-8B** predicts the **next GUI state as executable web code**. This approach ensures pixel-perfect text rendering and structurally accurate layouts, overcoming the hallucination and legibility issues common in pixel-generation models.
 
+This model was presented in the paper [Generative Visual Code Mobile World Models](https://huggingface.co/papers/2602.01576).
 
 <p align="center">
 <picture>
@@ -28,6 +31,114 @@ language:
 </picture>
 </p>
 
+## Sample Usage
+
+You can run inference using the `vLLM` library as follows:
+
+```python
+from vllm import LLM, SamplingParams
+from transformers import AutoProcessor
+from PIL import Image
+
+# Model configuration (choose one)
+# For gWorld-8B:
+MODEL_PATH = "trillionlabs/gWorld-8B"
+BASE_MODEL = "Qwen/Qwen3-VL-8B-Instruct"
+
+# For gWorld-32B:
+# MODEL_PATH = "trillionlabs/gWorld-32B"
+# BASE_MODEL = "Qwen/Qwen3-VL-32B"
+
+# Image processing settings
+MM_PROCESSOR_KWARGS = {
+    "max_pixels": 4233600,
+    "min_pixels": 3136,
+}
+
+# Load model
+llm = LLM(
+    model=MODEL_PATH,
+    tokenizer=BASE_MODEL,
+    tensor_parallel_size=8,
+    gpu_memory_utilization=0.9,
+    max_model_len=19384,
+    trust_remote_code=True,
+    mm_processor_kwargs=MM_PROCESSOR_KWARGS,
+    enable_chunked_prefill=True,
+    max_num_batched_tokens=16384,
+)
+
+# Load processor for chat template
+processor = AutoProcessor.from_pretrained(BASE_MODEL, trust_remote_code=True)
+
+# Prepare input
+image = Image.open("screenshot.png")
+if image.mode != 'RGB':
+    image = image.convert('RGB')
+
+action = '{"action_type": "TAP", "coordinates": [512, 890]}'
+
+# World model prompt template
+user_content = f"""You are an expert mobile UI World Model that can accurately predict the next state given an action.
+Given a screenshot of a mobile interface and an action, you must generate clean, responsive HTML code that represents the state of the interface AFTER the action is performed.
+First generate reasoning about what the next state should look like based on the action.
+Afterwards, generate the HTML code representing the next state that logically follows the action.
+You will render this HTML in a mobile viewport to see how similar it looks and acts like the mobile screenshot.
+
+Requirements:
+1. Provide reasoning about what the next state should look like based on the action
+2. Generate complete, valid HTML5 code
+3. Choose between using inline CSS and utility classes from Bootstrap, Tailwind CSS, or MUI for styling, depending on which option generates the closest code to the screenshot.
+4. Use mobile-first design principles matching screenshot dimensions.
+5. For images, use inline SVG placeholders with explicit width and height attributes that match the approximate dimensions from the screenshot. Matching the approximate color is also good.
+6. Use modern web standards and best practices
+7. Return ONLY the HTML code, no explanations or markdown formatting
+8. The generated HTML should render properly in a mobile viewport.
+9. Generated HTML should look like the screen that logically follows the current screen and the action.
+
+Action:
+{action}
+
+Output format:
+# Next State Reasoning: <your reasoning about what the next state should look like>
+# HTML: <valid_html_code>
+
+Generate the next state reasoning and the next state in html:"""
+
+# Build messages
+messages = [
+    {
+        "role": "user",
+        "content": [
+            {"type": "image", "image": image},
+            {"type": "text", "text": user_content},
+        ],
+    }
+]
+
+# Apply chat template
+prompt = processor.apply_chat_template(
+    messages,
+    tokenize=False,
+    add_generation_prompt=True,
+)
+
+# Generation parameters
+sampling_params = SamplingParams(
+    max_tokens=15000,
+    temperature=0,
+    seed=42,
+    top_p=1.0,
+)
+
+# Generate
+outputs = llm.generate(
+    [{"prompt": prompt, "multi_modal_data": {"image": image}}],
+    sampling_params=sampling_params
+)
+
+print(outputs[0].outputs[0].text)
+```
 
 ## Model Summary
 - **Architecture:** Based on `Qwen3-VL-8B`
@@ -70,12 +181,12 @@ This model is licensed under the Apache License 2.0. For inquiries, please conta
 ## Citation
 ```
 @misc{koh2026generativevisualcodemobile,
-      title={Generative Visual Code Mobile World Models},
-      author={Woosung Koh and Sungjun Han and Segyu Lee and Se-Young Yun and Jamin Shin},
-      year={2026},
-      eprint={2602.01576},
-      archivePrefix={arXiv},
-      primaryClass={cs.LG},
-      url={https://arxiv.org/abs/2602.01576},
+      title={Generative Visual Code Mobile World Models},
+      author={Woosung Koh and Sungjun Han and Segyu Lee and Se-Young Yun and Jamin Shin},
+      year={2026},
+      eprint={2602.01576},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG},
+      url={https://arxiv.org/abs/2602.01576},
 }
 ```