Peter commited on
Commit
d080a18
1 Parent(s): 146762d

update from basic sc

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -124,7 +124,6 @@ def ask_gpt(
124
  if basic_sc and not detect_propers(rawtxt):
125
  cln_resp = symspeller(rawtxt, sym_checker=schnellspell)
126
  elif not detect_propers(rawtxt):
127
- cln_resp = neuspell_correct(rawtxt, checker=ns_checker)
128
  cln_resp = synthesize_grammar(corrector=grammarbot, message=cln_resp)
129
  else:
130
  # no correction needed
@@ -165,7 +164,7 @@ def get_parser():
165
  parser.add_argument(
166
  "--basic-sc",
167
  required=False,
168
- default=True, # TODO: change this back to False once Neuspell issues are resolved.
169
  action="store_true",
170
  help="turn on symspell (baseline) correction instead of the more advanced neural net models",
171
  )
@@ -199,8 +198,7 @@ if __name__ == "__main__":
199
  print("Using the baseline spellchecker")
200
  schnellspell = build_symspell_obj()
201
  else:
202
- print("using Neuspell spell checker")
203
- ns_checker = load_ns_checker(fast=False)
204
  grammarbot = pipeline("'text2text-generation", gram_model, device=device)
205
 
206
  print(f"using model stored here: \n {model_loc} \n")
 
124
  if basic_sc and not detect_propers(rawtxt):
125
  cln_resp = symspeller(rawtxt, sym_checker=schnellspell)
126
  elif not detect_propers(rawtxt):
 
127
  cln_resp = synthesize_grammar(corrector=grammarbot, message=cln_resp)
128
  else:
129
  # no correction needed
 
164
  parser.add_argument(
165
  "--basic-sc",
166
  required=False,
167
+ default=False, # TODO: change this back to False once Neuspell issues are resolved.
168
  action="store_true",
169
  help="turn on symspell (baseline) correction instead of the more advanced neural net models",
170
  )
 
198
  print("Using the baseline spellchecker")
199
  schnellspell = build_symspell_obj()
200
  else:
201
+ print("using neural spell checker")
 
202
  grammarbot = pipeline("'text2text-generation", gram_model, device=device)
203
 
204
  print(f"using model stored here: \n {model_loc} \n")