Update README.md

README.md

@@ -4,14 +4,24 @@ tags:
 - SQL Injection
 ---
 
-**SQL Injection Attack Detection using a Fine-tuned distilBERT Model**
+**A distilBERT based SQL Injection Detection Model**
 
 **Model description**
 
-This model was created by finetuning the distilBERT model for SQL Injection Attack Detection.
-The model was fine-tuned using HuggingFace's Trainer API. The final result of the finetuning has an F1-score of 0.9986 and an accuracy of 99.90%.
-The model was finetuned with 10 epochs.
+This model, based on DistilBERT, is specifically tailored for the detection of SQL injection attacks. 
+Through fine-tuning using the Hugging Face's Trainer API, the model has been trained to identify potentially malicious SQL queries with high accuracy.
+
+- __Architecture:__ DistilBERT
+- __Fine-tuning Method:__ Trainer API
+- __Performance Metrics:__
+  - __F1-score:__ 0.9986
+  -  __Accuracy:__ 99.90%
+- __Training Epochs:__ 10
 
 **Dataset description**
 
-The finetuning was done on a SQL Injection Dataset that was made available by SAJID576 on Kaggle (https://www.kaggle.com/datasets/sajid576/sql-injection-dataset/data). It has a size of slightly over 30,000 rows.
+The model was fine-tuned on the SQL Injection Dataset, curated and made available by SAJID576 on Kaggle. 
+This dataset comprises of 30,920 rows of SQL queries, including both benign and malicious examples, providing a comprehensive training corpus for robust model development.
+
+- Dataset Source: https://www.kaggle.com/datasets/sajid576/sql-injection-dataset/data
+- Size: 30,920 rows