asd

app.py

@@ -208,9 +208,9 @@ classifier_tokenizer = AutoTokenizer.from_pretrained(classifier_model_name)
 classifier_model = AutoModelForSequenceClassification.from_pretrained(classifier_model_name).to(device)
 
 # Initialize grammar correction model and tokenizer
-grammar_model_name = "NoaiGPT/grammarly-coedit-large"
-grammar_tokenizer = AutoTokenizer.from_pretrained(grammar_model_name,use_auth_token=hf_token)
-grammar_model = T5ForConditionalGeneration.from_pretrained(grammar_model_name, use_auth_token=hf_token).to(device)
+grammar_model_name = "grammarly/coedit-large"
+grammar_tokenizer = AutoTokenizer.from_pretrained(grammar_model_name)
+grammar_model = T5ForConditionalGeneration.from_pretrained(grammar_model_name).to(device)
 
 # Initialize sentence splitter
 splitter = SentenceSplitter(language='en')
@@ -229,6 +229,7 @@ def correct_grammar(text):
     inputs = grammar_tokenizer(f'Fix grammatical errors in this sentence: {text}', return_tensors="pt").input_ids.to(device)
     outputs = grammar_model.generate(inputs, max_length=256)
     corrected_text = grammar_tokenizer.decode(outputs[0], skip_special_tokens=True)
+    print(corrected_text)
     return corrected_text
 
 @spaces.GPU