Update README.md

README.md

@@ -1,9 +1,37 @@
 ---
 license: cc-by-4.0
+base_model: jonfd/electra-base-igc-is
 language:
 - is
+datasets:
+- Sigurdur/imdb-isl-google-translate
+pipeline_tag: text-classification
 ---
 
-# Model
-This model is a fine-tuned version of [jonfd/electra-base-igc-is](https://huggingface.co/jonfd/electra-base-igc-is), for sentiment analysis. It was trained using [IMDb movie reviews](https://www.kaggle.com/datasets/lakshmi25npathi/imdb-dataset-of-50k-movie-reviews/data) translated into Icelandic using Google Translate.
+# Electra Icelandic Sentiment Analysis
 
+## Model Description
+
+This model is a fine-tuned version of [jonfd/electra-base-igc-is](https://huggingface.co/jonfd/electra-base-igc-is), for sentiment analysis. It was trained using a [IMDb movie reviews](https://www.kaggle.com/datasets/lakshmi25npathi/imdb-dataset-of-50k-movie-reviews/data) translated into Icelandic using Google Translate.
+
+## Model Sources
+
+- **Repository**: https://github.com/cadia-lvl/sentiment-analysis
+- **Paper**: https://skemman.is/bitstream/1946/46224/1/FinalReport.pdf
+
+## Citation
+
+**BibTeX:**
+
+```tex
+@techreport{Johannsson2023,
+  title = {Evaluating Icelandic Sentiment Analysis Models Trained on Translated Data},
+  author = {Ólafur Aron Johannsson and Birkir Arndal and Eysteinn Örn},
+  year = {2023},
+  institution = {University of Reykjavik},
+  department = {Department of Computer Science},
+  month = {12},
+  day = {15},
+  note = {Supervised by Stefán Ólafsson and Hrafn Loftsson, Examined by Sigurjón Ingi Garðarsson}
+}
+```