Update README.md

README.md

@@ -8,6 +8,11 @@ tags:
 - Fire-Detection-engine
 - Precision-98
 ---
+![uioashedfvg.png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/V9KmK7VrvuqyhD3S8eO7Z.png)
+
+# **Fire-Detection-Engine**
+
+The **Fire-Detection-Engine** is a state-of-the-art deep learning model designed to detect fire-related conditions in images. It leverages the **Vision Transformer (ViT)** architecture, specifically the `google/vit-base-patch16-224-in21k` model, fine-tuned on a dataset of fire and non-fire images. The model is trained to classify images into one of the following categories: "Fire Needed Action," "Normal Conditions," or "Smoky Environment," making it a powerful tool for detecting fire hazards.
 
 ```
 Classification report:
@@ -23,4 +28,21 @@ Fire Needed Action     0.9708    0.9864    0.9785       808
       weighted avg     0.9799    0.9798    0.9797      2424
 ```
 
-![download.png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/kDS5cVee2ZVOv92rY0lGw.png)
+![download.png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/kDS5cVee2ZVOv92rY0lGw.png)
+
+
+# **Mappers**
+
+```python
+Mapping of IDs to Labels: {0: 'Fire Needed Action', 1: 'Normal Conditions', 2: 'Smoky Environment'} 
+
+Mapping of Labels to IDs: {'Fire Needed Action': 0, 'Normal Conditions': 1, 'Smoky Environment': 2}
+```
+
+# **Key Features**
+- **Architecture**: Vision Transformer (ViT) - `google/vit-base-patch16-224-in21k`.
+- **Input**: RGB images resized to 224x224 pixels.
+- **Output**: Binary classification ("Fire Needed Action" or "Normal Conditions" or "Smoky Environment").
+- **Training Dataset**: A curated dataset of fire place conditions.
+- **Fine-Tuning**: The model is fine-tuned using Hugging Face's `Trainer` API with advanced data augmentation techniques.
+- **Performance**: Achieves high accuracy and F1 score on validation and test datasets.