Update app.py

app.py

@@ -6,7 +6,10 @@ from optimum.pipelines import pipeline
 
 def check_grammar(sentence):
   tokenizer = AutoTokenizer.from_pretrained("leslyarun/grammatical-error-correction-quantized")
-  model = ORTModelForSeq2SeqLM.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)