Update README.md

README.md

@@ -1,6 +1,6 @@
 ---
 library_name: transformers
-license: apache-2.0
+license: mit
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
@@ -12,6 +12,10 @@ metrics:
 model-index:
 - name: finetuned_model
   results: []
+datasets:
+- cassieli226/cities-text-dataset
+language:
+- en
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -29,15 +33,15 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+It is a finetuned model for binary classification of description of city. It will result in either Pittsburgh or Shanghai.
 
 ## Intended uses & limitations
 
-More information needed
+This is for education and demonstration purposes.
 
 ## Training and evaluation data
 
-More information needed
+The data for finetuning the model comes from this HF dataset: cassieli226/cities-text-dataset
 
 ## Training procedure
 
@@ -68,4 +72,4 @@ The following hyperparameters were used during training:
 - Transformers 4.56.1
 - Pytorch 2.8.0+cu126
 - Datasets 4.0.0
-- Tokenizers 0.22.0
+- Tokenizers 0.22.0