Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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())
|