Update README.md

README.md

@@ -5,6 +5,17 @@ language:
 - de
 base_model:
 - Unbabel/TowerInstruct-13B-v0.1
+tags:
+- Automatic Post Editing
+- Machine Translation
+model-index:
+  - name: iwslt_ape_ende
+    results: []
+    paper:
+      title: "KIT's Offline Speech Translation and Instruction Following Submission for IWSLT 2025"
+      authors: "Koneru, Sai and Z{\"u}fle, Maike and Nguyen, Thai-Binh and Akti, Seymanur and Niehues, Jan and Waibel, Alexander"
+      url: https://arxiv.org/abs/2505.13036
+      published: 2025-05-25
 ---
 
 # KIT IWSLT25 Automatic Post-Editing Model
@@ -58,4 +69,17 @@ num_beams=5
 output = model.generate(**inputs, num_beams=num_beams, max_new_tokens=256, return_dict_in_generate=True, early_stopping=True, do_sample=False)
 hyps = tokenizer.batch_decode(output.sequences[:,inputs.input_ids.shape[1]:], skip_special_tokens=True)
 print(hyps)
-```
+```
+
+## 📖 Citation
+
+If you use this model in your research, please cite:
+
+```bibtex
+@inproceedings{koneru2025kit,
+  title={KIT's Offline Speech Translation and Instruction Following Submission for IWSLT 2025},
+  author={Koneru, Sai and Z{\"u}fle, Maike and Nguyen, Thai-Binh and Akti, Seymanur and Niehues, Jan and Waibel, Alexander},
+  journal={arXiv preprint arXiv:2505.13036},
+  year={2025},
+  url={https://arxiv.org/abs/2505.13036}
+}