Балаганский Никита Николаевич
commited on
Commit
•
a14f488
1
Parent(s):
a905c49
add citation info
Browse files
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:
|