File size: 1,100 Bytes
25c77d0
 
 
 
 
 
 
 
 
 
ce710c5
 
 
 
 
 
 
 
 
 
 
86c41d1
 
 
ce710c5
 
 
 
90e4d55
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
---
language: en
tags:
- grammar
- text2text-generation
datasets:
- leslyarun/c4_200m_gec_train100k_test25k
---

# Get Grammatical corrections on your English text, trained on a subset of c4-200m dataset - ONNX Quantized Model

# Use the below code for running the model

``` python

from transformers import AutoTokenizer
from optimum.onnxruntime import ORTModelForSeq2SeqLM
from optimum.pipelines import pipeline

tokenizer = AutoTokenizer.from_pretrained("leslyarun/grammatical-error-correction-quantized")
model = ORTModelForSeq2SeqLM.from_pretrained("leslyarun/grammatical-error-correction-quantized",
                                              encoder_file_name="encoder_model_quantized.onnx",
                                              decoder_file_name="decoder_model_quantized.onnx",
                                              decoder_with_past_file_name="decoder_with_past_model_quantized.onnx")
  
text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
output = text2text_generator("grammar: " + sentence)
print(output[0]["generated_text"])

```