Peter commited on
Commit
6b6c220
1 Parent(s): 6e1a316
Files changed (2) hide show
  1. app.py +1 -1
  2. grammar_improve.py +2 -0
app.py CHANGED
@@ -201,7 +201,7 @@ if __name__ == "__main__":
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")
207
  iface = gr.Interface(
 
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")
207
  iface = gr.Interface(
grammar_improve.py CHANGED
@@ -109,6 +109,7 @@ def fix_punct_spacing(text: str):
109
 
110
  return cln_text
111
 
 
112
  def synthesize_grammar(
113
  corrector: transformers.pipeline,
114
  message: str,
@@ -154,6 +155,7 @@ def synthesize_grammar(
154
  print(f"synthesizing took {rt} seconds")
155
  return corrected_text.strip()
156
 
 
157
  """
158
  start of SymSpell code
159
  """
 
109
 
110
  return cln_text
111
 
112
+
113
  def synthesize_grammar(
114
  corrector: transformers.pipeline,
115
  message: str,
 
155
  print(f"synthesizing took {rt} seconds")
156
  return corrected_text.strip()
157
 
158
+
159
  """
160
  start of SymSpell code
161
  """