Vivek12345323 commited on
Commit
f304fac
·
verified ·
1 Parent(s): 71e359d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ pipe = pipeline("text2text-generation", model="pszemraj/flan-t5-large-grammar-sy
7
 
8
  def respond(prompt):
9
  # Generate revised essay
10
- revised = pipe(prompt, max_new_tokens=512)[0]['generated_text']
11
 
12
  # Generate a word-level diff
13
  diff = difflib.ndiff(prompt.split(), revised.split())
 
7
 
8
  def respond(prompt):
9
  # Generate revised essay
10
+ revised = pipe(prompt, max_new_tokens=1024)[0]['generated_text']
11
 
12
  # Generate a word-level diff
13
  diff = difflib.ndiff(prompt.split(), revised.split())