asd
Browse files
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
|
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 |
|