Балаганский Никита Николаевич commited on
Commit
a14f488
1 Parent(s): a905c49

add citation info

Browse files
Files changed (1) hide show
  1. app.py +22 -0
app.py CHANGED
@@ -33,6 +33,25 @@ ATTRIBUTE_MODELS = {
33
  )
34
  }
35
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
36
  LANGUAGE_MODELS = {
37
  "Russian": (
38
  'sberbank-ai/rugpt3small_based_on_gpt2',
@@ -194,6 +213,9 @@ def main():
194
 
195
  st.button("Generate new", on_click=generate())
196
 
 
 
 
197
 
198
  @st.cache(hash_funcs={tokenizers.Tokenizer: lambda lm_tokenizer: hash(lm_tokenizer.to_str)}, allow_output_mutation=True)
199
  def load_generator(lm_model_name: str) -> Generator:
 
33
  )
34
  }
35
 
36
+ CITE = """@misc{https://doi.org/10.48550/arxiv.2205.07276,
37
+ doi = {10.48550/ARXIV.2205.07276},
38
+
39
+ url = {https://arxiv.org/abs/2205.07276},
40
+
41
+ author = {Sitdikov, Askhat and Balagansky, Nikita and Gavrilov, Daniil and Markov, Alexander},
42
+
43
+ keywords = {Computation and Language (cs.CL), Machine Learning (cs.LG), FOS: Computer and information sciences, FOS: Computer and information sciences},
44
+
45
+ title = {Classifiers are Better Experts for Controllable Text Generation},
46
+
47
+ publisher = {arXiv},
48
+
49
+ year = {2022},
50
+
51
+ copyright = {Creative Commons Attribution 4.0 International}
52
+ }
53
+ """
54
+
55
  LANGUAGE_MODELS = {
56
  "Russian": (
57
  'sberbank-ai/rugpt3small_based_on_gpt2',
 
213
 
214
  st.button("Generate new", on_click=generate())
215
 
216
+ st.subheader("Citation")
217
+ st.code(CITE)
218
+
219
 
220
  @st.cache(hash_funcs={tokenizers.Tokenizer: lambda lm_tokenizer: hash(lm_tokenizer.to_str)}, allow_output_mutation=True)
221
  def load_generator(lm_model_name: str) -> Generator: