Update README.md

README.md

@@ -1,6 +1,14 @@
 ---
 base_model: meta-llama/Meta-Llama-3-8B
 library_name: peft
+license: mit
+language:
+- en
+metrics:
+- accuracy
+- precision
+- recall
+- f1
 ---
 
 ## Llama3 8B Fine-Tuned for Domain Generation Algorithm Detection
@@ -23,7 +31,7 @@ This model leverages the extensive semantic understanding of Llama3 to classify
 
 The model was trained with 2.5 million domains, split between 1.5 million DGA domains and 1 million normal domains. 
 
-Data Link: https://github.com/hmaccelerate/DGA_Detection/blob/master/data/mixed_domain.csv
+Dataset Link: https://huggingface.co/datasets/AbuSayed1/DGA-DATASET
 
 The GitHub repository https://github.com/MDABUSAYED/StratosphereLinuxIPS/tree/LLM/modules/flowalerts/LLM describe how the model was trained and evaluated.
 
@@ -42,5 +50,4 @@ The GitHub repository https://github.com/MDABUSAYED/StratosphereLinuxIPS/tree/LL
 
 **APA:**
 
-Sayed, M. A., Rahman, A., Kiekintveld, C., & Garcia, S. (2024). Fine-tuning Large Language Models for DGA and DNS Exfiltration Detection. arXiv preprint arXiv:2410.21723.
-
+Sayed, M. A., Rahman, A., Kiekintveld, C., & Garcia, S. (2024). Fine-tuning Large Language Models for DGA and DNS Exfiltration Detection. arXiv preprint arXiv:2410.21723.