Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ import tokenizers
|
|
14 |
#os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
15 |
|
16 |
random.seed(None)
|
17 |
-
suggested_text_list = ['
|
18 |
|
19 |
@st.cache(hash_funcs={tokenizers.Tokenizer: id, tokenizers.AddedToken: id})
|
20 |
def load_model(model_name):
|
@@ -95,13 +95,13 @@ if __name__ == "__main__":
|
|
95 |
|
96 |
model.to(device)
|
97 |
|
98 |
-
text_area = st.text_area("Enter the first few words (or leave blank), tap on \"Generate Text\" below. Tapping again will produce a different result.", '
|
99 |
|
100 |
st.sidebar.subheader("Configurable parameters")
|
101 |
|
102 |
-
max_len = st.sidebar.slider("Max-Length", 0, 256,
|
103 |
-
top_k = st.sidebar.slider("Top-K", 0, 100,
|
104 |
-
top_p = st.sidebar.slider("Top-P", 0.0, 1.0, 0
|
105 |
|
106 |
if st.button("Generate Text"):
|
107 |
with st.spinner(text="Generating results..."):
|
|
|
14 |
#os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
15 |
|
16 |
random.seed(None)
|
17 |
+
suggested_text_list = ['诪讟专转 讛讘专讬讗讛']
|
18 |
|
19 |
@st.cache(hash_funcs={tokenizers.Tokenizer: id, tokenizers.AddedToken: id})
|
20 |
def load_model(model_name):
|
|
|
95 |
|
96 |
model.to(device)
|
97 |
|
98 |
+
text_area = st.text_area("Enter the first few words (or leave blank), tap on \"Generate Text\" below. Tapping again will produce a different result.", '诪讟专转 讛讘专讬讗讛')
|
99 |
|
100 |
st.sidebar.subheader("Configurable parameters")
|
101 |
|
102 |
+
max_len = st.sidebar.slider("Max-Length", 0, 256, 80, help="The maximum length of the sequence to be generated.")
|
103 |
+
top_k = st.sidebar.slider("Top-K", 0, 100, 0, help="The number of highest probability vocabulary tokens to keep for top-k-filtering.")
|
104 |
+
top_p = st.sidebar.slider("Top-P", 0.0, 1.0, 1.0, help="If set to float < 1, only the most probable tokens with probabilities that add up to top_p or higher are kept for generation.")
|
105 |
|
106 |
if st.button("Generate Text"):
|
107 |
with st.spinner(text="Generating results..."):
|