Upload folder using huggingface_hub

README.md

@@ -0,0 +1,95 @@
+---
+license: apache-2.0
+base_model: google/vit-base-patch16-224
+tags:
+- Image Regression
+datasets:
+- "tonyassi/revolve-6-17-24"
+metrics:
+- accuracy
+model-index:
+- name: "tonyassi/revolve-6-17-24-prediction"
+  results: []
+---
+
+# tonyassi/revolve-6-17-24-prediction
+## Image Regression Model
+
+This model was trained with [Image Regression Model Trainer](https://github.com/TonyAssi/ImageRegression/tree/main). It takes an image as input and outputs a float value.
+
+```python
+from ImageRegression import predict
+predict(repo_id='tonyassi/revolve-6-17-24-prediction',image_path='image.jpg')
+```
+
+---
+
+## Dataset
+Dataset: tonyassi/revolve-6-17-24\
+Value Column: 'sales_index'\
+Train Test Split: 0.2
+
+---
+
+## Training
+Base Model: [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224)\
+Epochs: 10\
+Learning Rate: 0.0001
+
+---
+
+## Usage
+
+### Download
+```bash
+git clone https://github.com/TonyAssi/ImageRegression.git
+cd ImageRegression
+```
+
+### Installation
+```bash
+pip install -r requirements.txt
+```
+
+### Import 
+```python
+from ImageRegression import train_model, upload_model, predict
+```
+
+### Inference (Prediction)
+- **repo_id** 🤗 repo id of the model
+- **image_path** path to image
+```python
+predict(repo_id='tonyassi/revolve-6-17-24-prediction',
+        image_path='image.jpg')
+```
+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
+- **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
+- **num_train_epochs** training epochs
+- **learning_rate** learning rate
+```python
+train_model(dataset_id='tonyassi/revolve-6-17-24',
+            value_column_name='sales_index',
+            test_split=0.2,
+            output_dir='./results',
+            num_train_epochs=10,
+            learning_rate=0.0001)
+
+```
+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='tonyassi/revolve-6-17-24-prediction',
+             token='YOUR_HF_TOKEN',
+             checkpoint_dir='./results/checkpoint-940')
+```

metadata.json

@@ -0,0 +1,8 @@
+{
+    "dataset_id": "tonyassi/revolve-6-17-24",
+    "value_column_name": "sales_index",
+    "test_split": 0.2,
+    "num_train_epochs": 10,
+    "learning_rate": 0.0001,
+    "max_value": 14954
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c5a76ab3445a07f9d4f6aa46e67a988bafce9c3a9fccd578831545bd68f9cdd
+size 345583444

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:875a54da3ff1224ecf71e153ddbfa892d770cf201fb6bf5fc5510db600dc0444
+size 686557178

rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da1db5c227c2000e391e1d225e13a38eda71746be2164bab198c44af9ae0882b
+size 13990

scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c795f7641ddf11bcb947b666516063a2b7b5ecf7517e24703919e1f96977ecda
+size 1064

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8e6f6c4db76f70b7c87c96d767deafe46bcf5e261de3d701ff35a9b7f8d06529
+size 5048