Add link to Github repository

README.md

@@ -17,6 +17,8 @@ DuoGuard-0.5B is built upon Qwen 2.5 (0.5B), a multilingual large language model
 
 -It is presented in the paper [DuoGuard: A Two-Player RL-Driven Framework for Multilingual LLM Guardrails](https://huggingface.co/papers/2502.05163)
 
+For code, see https://github.com/yihedeng9/DuoGuard.
+
 ## How to Use
 A quick code snippet or set of instructions on how to load and use the model in an application:
 ```python
@@ -72,7 +74,7 @@ category_names = [
 prob_vector = probabilities[0].tolist()  # shape: (12,)
 
 predicted_labels = []
-for cat_name, prob in zip(category_names, prob_vector):
+for cat_name, prob, label in zip(category_names, prob_vector):
     label = 1 if prob > threshold else 0
     predicted_labels.append(label)