Upload folder using huggingface_hub

README.md

@@ -29,7 +29,10 @@ model-index:
 
 # Cheese Texture Classifier (AutoML)
 
-This model performs 4-class texture classification on cheese images using AutoML-optimized transfer learning.
+**Model Creator**: Rumi Loghmani (@rlogh)  
+**Original Dataset**: aslan-ng/cheese-image (by Aslan Noorghasemi)
+
+This model performs 4-class texture classification on cheese images using AutoML-optimized transfer learning. The model was developed by Rumi Loghmani using the cheese image dataset created by Aslan Noorghasemi.
 
 ## Model Description
 
@@ -40,7 +43,8 @@ This model performs 4-class texture classification on cheese images using AutoML
 
 ## Training Details
 
-- **Dataset**: [aslan-ng/cheese-image](https://huggingface.co/datasets/aslan-ng/cheese-image)
+- **Model Developer**: Rumi Loghmani (@rlogh)
+- **Dataset**: [aslan-ng/cheese-image](https://huggingface.co/datasets/aslan-ng/cheese-image) (by Aslan Noorghasemi)
 - **AutoML Method**: Optuna with 20 trials
 - **Transfer Learning**: Pre-trained resnet34 backbone
 - **Early Stopping**: Yes (patience=10)
@@ -122,8 +126,19 @@ print(f"Predicted class: {class_names[predicted_class]}")
 
 ## Citation
 
-If you use this model, please cite the original dataset:
+If you use this model, please cite both the model and the original dataset:
+
+**Model Citation:**
+```bibtex
+@model{rlogh/cheese-texture-classifier-automl,
+  title={Cheese Texture Classifier (AutoML)},
+  author={Rumi Loghmani},
+  year={2024},
+  url={https://huggingface.co/rlogh/cheese-texture-classifier-automl}
+}
+```
 
+**Dataset Citation:**
 ```bibtex
 @dataset{aslan-ng/cheese-image,
   title={Cheese Image Dataset},