pszemraj commited on
Commit
896d60b
1 Parent(s): 9b88e87

:zap: update grammar model to base, update temp default

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -154,7 +154,7 @@ def get_parser():
154
  "--gram-model",
155
  required=False,
156
  type=str,
157
- default="pszemraj/grammar-synthesis-small",
158
  help="text2text generation model ID from huggingface for the model to correct grammar",
159
  )
160
 
@@ -209,7 +209,7 @@ if __name__ == "__main__":
209
  lines=2,
210
  ),
211
  Slider(
212
- minimum=0.0, maximum=1.0, step=0.01, default=0.6, label="temperature"
213
  ),
214
  Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="top_p"),
215
  Slider(minimum=0, maximum=100, step=5, default=20, label="top_k"),
 
154
  "--gram-model",
155
  required=False,
156
  type=str,
157
+ default="pszemraj/grammar-synthesis-base",
158
  help="text2text generation model ID from huggingface for the model to correct grammar",
159
  )
160
 
 
209
  lines=2,
210
  ),
211
  Slider(
212
+ minimum=0.0, maximum=1.0, step=0.05, default=0.4, label="temperature"
213
  ),
214
  Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="top_p"),
215
  Slider(minimum=0, maximum=100, step=5, default=20, label="top_k"),