Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,33 @@
+
+---
+tags:
+- image-classification
+- wildfire-detection
+datasets:
+- your-dataset-name # Optional: Add the name of the dataset you used
+---
+
+# ResNet50 Fine-tuned for Wildfire Detection
+
+This model is a ResNet50 fine-tuned on the [Wildfire Prediction Dataset](https://www.kaggle.com/datasets/abdelghaniaaba/wildfire-prediction-dataset) for image classification to detect wildfires.
+
+## Model Details
+
+The model is a ResNet50 architecture with a custom classification head. It was trained in two phases: first with the backbone frozen, and then with full fine-tuning.
+
+## Training
+
+The model was trained on the provided dataset with the following characteristics:
+- **Train samples:** 30250
+- **Validation samples:** 6300
+- **Batch size:** 16
+- **Epochs (Phase 1 - Head):** 5
+- **Epochs (Phase 2 - Full Fine-tuning):** 3
+
+## Performance
+
+(Add details about the model's performance metrics, e.g., accuracy, loss, etc. from your training output)
+
+## Usage
+
+(Provide code examples on how to load and use the model for inference)