Update README.md

README.md

@@ -1,22 +1,36 @@
----
-language:
-- en
-license: apache-2.0
-tags:
-- text-generation-inference
-- transformers
-- unsloth
-- mistral
-- trl
-base_model: unsloth/mistral-7b-bnb-4bit
----
-
-# Uploaded  model
+---
+language:
+- en
+license: apache-2.0
+tags:
+- text-generation-inference
+- transformers
+- unsloth
+- mistral
+- trl
+- python
+base_model: unsloth/mistral-7b-bnb-4bit
+---
+
+# AI-DocGen Model
+
 
 - **Developed by:** eltaiefaymen
 - **License:** apache-2.0
 - **Finetuned from model :** unsloth/mistral-7b-bnb-4bit
 
-This mistral model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
+# Overview
+This model, AI-DocGen, is a finetuned version of the unsloth/mistral-7b-bnb-4bit model. It has been specifically trained to generate docstrings for Python code snippets according to Python's docstring conventions.
+
+# Features
+- **Fast Finetuning:** The model was trained 2x faster using Unsloth and Huggingface's TRL library.
+- **Optimized for Documentation:** Generates comprehensive and accurate docstrings for Python functions and classes.
+- **Efficient Inference:** Supports 4-bit quantization for reduced memory usage and faster inference.
+
+# Training Details
+- **Base Model:** unsloth/mistral-7b-bnb-4bit
+- **Training Library:** Huggingface's TRL
+- **Training Framework:** Unsloth
+
 
-[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
+[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)