tweaks to model card

.gitattributes

@@ -30,3 +30,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,10 +1,12 @@
 ---
-license: unknown
-language:
-  - en
+license: apache-2.0
 tags:
   - mlconsole
+  - tabular-regression
 library_name: mlconsole
+inference: false
+datasets:
+- julien-c/kaggle-rounakbanik-pokemon
 metrics:
   - mae
   - loss
@@ -12,10 +14,10 @@ model-index:
   - name: pokemon-predict-hp
     results:
       - task:
-          type: regression
-          name: regression
+          type: tabular-regression
+          name: tabular-regression
         dataset:
-          type: pokemon.csv
+          type: julien-c/kaggle-rounakbanik-pokemon
           name: pokemon.csv
         metrics:
           - type: mae
@@ -28,6 +30,14 @@ model-index:
 ---
 
 # pokemon.csv (#0)
-Trained on [ML Console](https://mlconsole.com).
+Trained on [ML Console](https://mlconsole.com) on the [julien-c/kaggle-rounakbanik-pokemon](https://huggingface.co/datasets/julien-c/kaggle-rounakbanik-pokemon).
 
 [Load the model on ML Console](https://mlconsole.com/model/hf/julien-c/pokemon-predict-hp).
+
+
+### Screenshots of training
+
+![](screenshots/training-curve.png)
+
+![](screenshots/predict.png)
+