ColeGuion commited on
Commit
1b691d8
·
verified ·
1 Parent(s): 911ff06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,8 +10,9 @@ model_name = "vennify/t5-base-grammar-correction"
10
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
 
13
- def correct_text(text, max_length, max_new_tokens=0, min_length, num_beams, temperature, top_p):
14
  inputs = tokenizer.encode(text, return_tensors="pt")
 
15
  if max_new_tokens > 0:
16
  outputs = model.generate(
17
  inputs,
 
10
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
 
13
+ def correct_text(text, max_length, max_new_tokens, min_length, num_beams, temperature, top_p):
14
  inputs = tokenizer.encode(text, return_tensors="pt")
15
+
16
  if max_new_tokens > 0:
17
  outputs = model.generate(
18
  inputs,