Update README.md

README.md

@@ -1,3 +1,15 @@
+# DeCLUTR-small
+
+## Model description
+
+The corresponding model from our paper: [DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representations](https://arxiv.org/abs/2006.03659).
+
+## Intended uses & limitations
+
+The model is intended to be used as a universal sentence encoder, similar to [Google's Universal Sentence Encoder](https://tfhub.dev/google/universal-sentence-encoder/4) or [Sentence Transformers](https://github.com/UKPLab/sentence-transformers).
+
+#### How to use
+
 ```python
 import torch
 from scipy.spatial.distance import cosine
@@ -26,4 +38,16 @@ embeddings = torch.sum(
 
 # Compute a semantic similarity via the cosine distance
 semantic_sim = 1 - cosine(embeddings[0], embeddings[1])
+```
+
+### BibTeX entry and citation info
+
+```bibtex
+@article{Giorgi2020DeCLUTRDC,
+  title={DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representations},
+  author={John M Giorgi and Osvald Nitski and Gary D. Bader and Bo Wang},
+  journal={ArXiv},
+  year={2020},
+  volume={abs/2006.03659}
+}
 ```