Add link to code

README.md

@@ -69,7 +69,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)
 
@@ -88,6 +88,8 @@ print(f"\nMaximum probability across all categories: {max_prob:.3f}")
 print(f"Overall Prompt Classification => {'UNSAFE' if overall_label == 1 else 'SAFE'}")
 ```
 
+You can find the code at https://github.com/yihedeng9/DuoGuard.
+
 ### Citation
 
 ```plaintext