Update README.md

README.md

@@ -7,9 +7,28 @@ tags:
 - transformers
 license: apache-2.0
 library_name: transformers
-pipeline_tag: image-regression
+pipeline_tag: image-classification
+model-index:
+- name: Image Quality Regression Model
+  results: []
 ---
 
+# Image Quality Regression Model
+
+This model is trained on the dataset [yigagilbert/image_quality_dataset](https://huggingface.co/datasets/yigagilbert/image_quality_dataset) and performs regression tasks to predict image quality scores. 
+
+## Model Details
+
+- **Dataset**: yigagilbert/image_quality_dataset
+- **Target Column**: quality_score
+- **Test Split**: 20% test data
+- **Training Epochs**: 3
+- **Learning Rate**: 5e-5
+- **Max Value in Dataset**: 54.02
+
+This model fine-tunes the [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) Vision Transformer using PyTorch and Hugging Face's 🤗 Transformers library. It predicts a numerical score based on the quality of the input image.
+
+
 # Image Regression Model
 
 This repository contains a model for **image regression** tasks, where the goal is to predict a numerical value from an input image. The model fine-tunes the [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) Vision Transformer using PyTorch and 🤗 Hugging Face tools. You can train the model, upload it to the 🤗 Model Hub, and perform inference using a simple API.