orendar commited on
Commit
4a92dd6
1 Parent(s): 7c6b8f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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, 192,help="The maximum length of the sequence to be generated.")
103
- top_k = st.sidebar.slider("Top-K", 0, 100, 40, 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, 0.92, 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..."):
 
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..."):