Update README.md

README.md

@@ -1,6 +1,9 @@
 ---
 library_name: transformers
-tags: []
+datasets:
+- tner/mit_movie_trivia
+base_model:
+- distilbert/distilbert-base-uncased
 ---
 
 # Model Card for Model ID
@@ -111,7 +114,12 @@ Use the code below to get started with the model.
 [More Information Needed]
 
 ## Training Details
-
+Training Details
+Optimizer: AdamW
+Learning Rate: 2e-5
+Batch Size: 8
+Evaluation Strategy: Epoch-based
+Loss Function: CrossEntropyLoss
 ### Training Data
 
 <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
@@ -162,16 +170,18 @@ Use the code below to get started with the model.
 [More Information Needed]
 
 ### Results
+Which movies did Quentin Tarantino make??
+Ans: Entity: quentin tarantino, Label: Director, Score: 1.00
 
-[More Information Needed]
-
+Leonardo DiCaprio starred in which movies?
+Ans: Entity: leonardo dicaprio, Label: Actor, Score: 1.00
 #### Summary
 
 
 
 ## Model Examination [optional]
 
-<!-- Relevant interpretability work for the model goes here -->
+
 
 [More Information Needed]