| from transformers import pipeline | |
| # Load the grammar correction model | |
| grammar_correction_model = pipeline(task="text2text-generation", model="hassaanik/grammar-correction-model") | |
| def correct_text(text): | |
| result = grammar_correction_model(text, max_length=200, num_beams=5, no_repeat_ngram_size=2) | |
| return result[0]['generated_text'] | |