Update README.md

README.md

@@ -3,14 +3,13 @@ library_name: transformers
 language:
 - en
 ---
-
-# Model Card for Model ID
+### Emotion Prediction for Written Stories
 
 <!-- Provide a quick summary of what the model is/does. -->
 
 This model is intended to predict emotions (valence, arousal) in written stories. For all details see [the paper (TODO)](#) and [the accompanying github repo (TODO)](#).
 
-## Model Details
+
 
 ### Model Description
 
@@ -35,6 +34,7 @@ We provide the best out of 5 seeds for each context size. Hence, the numbers in
 Technically, this model predicts token-wise valence/arousal values. Sentences are concatenated via the ``<s>`` token, where the valence/arousal predictions for an ``<s>`` token
 are meant to be the predictions for the sentence preceding it. All other tokens' predictions should be ignored. For reference, see the figure in the paper:
 
+![image](tales_vertical.png)
 
 The [accompanying repo](TODO) provides a convenient script to use the model for prediction.
 
@@ -42,8 +42,8 @@ The [accompanying repo](TODO) provides a convenient script to use the model for
 
 <!-- Provide the basic links for the model. -->
 
-- **Repository:** [accompanying repo](TODO)
-- **Paper:** [accompanying repo](TODO)
+- **Repository:** [Github](TODO)
+- **Paper:** [ArXiv](TODO)
 - 
 
 ## Uses
@@ -75,24 +75,6 @@ Please see the *Limitations* section in [the paper](#). Please note that the mod
 
 [More Information Needed]
 
-**APA:**
-
-[More Information Needed]
-
-## Glossary [optional]
-
-<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
-
-[More Information Needed]
-
-## More Information [optional]
-
-[More Information Needed]
-
-## Model Card Authors [optional]
-
-[More Information Needed]
-
 ## Model Card Contact
 
-[More Information Needed]
+For further inquiries, please contact lukas1[dot]christ[at]uni-a[dot].de