peter szemraj commited on
Commit
0e31f91
1 Parent(s): 88b1e11

streamline/fix params

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -117,10 +117,7 @@ if __name__ == "__main__":
117
  with demo:
118
  gr.Markdown("# Autocompleting Emails with Textgen - Demo")
119
  gr.Markdown(
120
- "Enter part of an email, and the model will autocomplete it for you!"
121
- )
122
- gr.Markdown(
123
- "The model used is [postbot/distilgpt2-emailgen](https://huggingface.co/postbot/distilgpt2-emailgen)"
124
  )
125
  gr.Markdown("---")
126
 
@@ -137,7 +134,7 @@ if __name__ == "__main__":
137
  )
138
  num_gen_tokens = gr.Slider(
139
  label="Generation Tokens",
140
- default=64,
141
  maximum=128,
142
  minimum=32,
143
  step=16,
@@ -153,7 +150,7 @@ if __name__ == "__main__":
153
  value=2,
154
  )
155
  length_penalty = gr.Slider(
156
- minimum=0.5, maximum=1.0, label="length penalty", default=0.8, step=0.05
157
  )
158
  generated_email = gr.Textbox(
159
  label="Generated Result",
 
117
  with demo:
118
  gr.Markdown("# Autocompleting Emails with Textgen - Demo")
119
  gr.Markdown(
120
+ "Enter part of an email, and the model will autocomplete it for you! The model used is [postbot/distilgpt2-emailgen](https://huggingface.co/postbot/distilgpt2-emailgen)"
 
 
 
121
  )
122
  gr.Markdown("---")
123
 
 
134
  )
135
  num_gen_tokens = gr.Slider(
136
  label="Generation Tokens",
137
+ value=64,
138
  maximum=128,
139
  minimum=32,
140
  step=16,
 
150
  value=2,
151
  )
152
  length_penalty = gr.Slider(
153
+ minimum=0.5, maximum=1.0, label="length penalty", value=0.8, step=0.1
154
  )
155
  generated_email = gr.Textbox(
156
  label="Generated Result",