Update README.md

README.md

@@ -1,51 +1,45 @@
 ---
 language:
 - en
-license: apache-2.0
+license: lgpl-3.0
+library_name: pytorch
 tags:
 - text-classification
 - transformers
 - pytorch
 - generated_from_keras_callback
+metrics:
+- accuracy
+- f1
 datasets:
 - m-newhauser/senator-tweets
-model-index:
-- name: distilbert-political-tweets
-  results: []
+widget:
+- text: "This pandemic has shown us clearly the vulgarity of our healthcare system. Highest costs in the world, yet not enough nurses or doctors. Many millions uninsured, while insurance company profits soar. The struggle continues. Healthcare is a human right. Medicare for all."
+  example_title: "Bernie Sanders (D)"
+- text: "Team Biden would rather fund the Ayatollah's Death to America regime than allow Americans to produce energy for our own domestic consumption."
+  example_title: "Ted Cruz (R)"
 ---
 
-<!-- This model card has been generated automatically according to the information Keras had access to. You should
-probably proofread and complete it, then remove this comment. -->
+# distilbert-political-tweets 🗣 🇺🇸
 
-# distilbert-political-tweets
-
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the m-newhauser/senator-tweets dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the [m-newhauser/senator-tweets](https://huggingface.co/datasets/m-newhauser/senator-tweets) dataset, which contains all tweets made by United States senators during the first year of the Biden Administration.
 It achieves the following results on the evaluation set:
-
+* Accuracy: 0.9076
+* F1: 0.9117
 
 ## Model description
 
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
+The goal of this model is to classify short pieces of text as having either Democratic or Republican sentiment. The model was fine-tuned on 99,693 tweets (51.6% Democrat, 48.4% Republican) made by US senators in 2021.
 
-More information needed
-
-## Training procedure
+Model accuracy may not hold up on pieces of text longer than a tweet.
 
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: None
+- optimizer: Adam
 - training_precision: float32
-
-### Training results
-
-
+- learning_rate = 5e-5
+- num_epochs = 5
 
 ### Framework versions