Update README.md

README.md

@@ -66,7 +66,7 @@ predict(repo_id='-',
 The first time this function is called it'll download the safetensor model. Subsequent function calls will run faster.
 
 ### Train Model
-- **dataset_id** 🤗 dataset id (see [Dataset](https://github.com/TonyAssi/ImageRegression?tab=readme-ov-file#dataset))
+- **dataset_id** 🤗 dataset id
 - **value_column_name** column name of prediction values in dataset
 - **test_split** test split of the train/test split
 - **output_dir** the directory where the checkpoints will be saved
@@ -83,3 +83,13 @@ train_model(dataset_id='-',
 ```
 The trainer will save the checkpoints in the output_dir location. The model.safetensors are the trained weights you'll use for inference (predicton).
 
+### Upload Model
+This function will upload your model to the 🤗 Hub.
+- **model_id** the name of the model id
+- **token** go [here](https://huggingface.co/settings/tokens) to create a new 🤗 token
+- **checkpoint_dir** checkpoint folder that will be uploaded
+```python
+upload_model(model_id='-',
+             token='YOUR_HF_TOKEN',
+             checkpoint_dir='./results/checkpoint-940')
+```