Update README.md

README.md

@@ -9,6 +9,8 @@ tags:
 - trl
 licence: license
 pipeline_tag: text-classification
+language:
+- en
 ---
 
 # Model Card for transformer-facebook-MobileLLM-R1-140M-ml-512-bs-32-ws-100-lr-1e-05-full_ft-merge_user_input_system_prompt
@@ -28,14 +30,11 @@ User input: I want to
 """
 generator = pipeline("text-generation", model="None", device="cuda")
 output = generator([{"role": "user", "content": question}], max_new_tokens=1, return_full_text=False)[0]
-print(output["generated_text"])
+print(output["generated_text"]) # "end" or "continue"
 ```
 
 ## Training procedure
 
-[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/dangvansam98/LLM-TURN-DETECTION/runs/ub528cqr) 
-
-
 This model was trained with SFT.
 
 ### Framework versions