NoaiGPT commited on
Commit
1ad5767
·
1 Parent(s): 0363ba6
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -208,7 +208,7 @@ classifier_tokenizer = AutoTokenizer.from_pretrained(classifier_model_name)
208
  classifier_model = AutoModelForSequenceClassification.from_pretrained(classifier_model_name).to(device)
209
 
210
  # Initialize grammar correction model and tokenizer
211
- grammar_model_name = "grammarly/coedit-large"
212
  grammar_tokenizer = AutoTokenizer.from_pretrained(grammar_model_name)
213
  grammar_model = T5ForConditionalGeneration.from_pretrained(grammar_model_name).to(device)
214
 
 
208
  classifier_model = AutoModelForSequenceClassification.from_pretrained(classifier_model_name).to(device)
209
 
210
  # Initialize grammar correction model and tokenizer
211
+ grammar_model_name = "NoaiGPT/grammarly-coedit-large"
212
  grammar_tokenizer = AutoTokenizer.from_pretrained(grammar_model_name)
213
  grammar_model = T5ForConditionalGeneration.from_pretrained(grammar_model_name).to(device)
214