Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import torch
 import spacy
 import subprocess
@@ -29,7 +29,7 @@ tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst
 model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english").to(device)
 
 # Load grammar correction model from Hugging Face
-grammar_corrector = pipeline("text2text-generation", model="vennify/t5-base-grammar-correction", device=0 if torch.cuda.is_available() else -1)
+grammar_corrector = pipeline("text2text-generation", model="pszemraj/flan-t5-large-grammar-synthesis", device=0 if torch.cuda.is_available() else -1)
 
 # AI detection function using DistilBERT
 def detect_ai_generated(text):
@@ -78,7 +78,7 @@ def paraphrase_with_spacy_nltk(text):
     
     return paraphrased_sentence
 
-# Grammar correction function using the Hugging Face model
+# Grammar correction function using FLAN-T5
 def correct_grammar(text):
     corrected_text = grammar_corrector(text)[0]['generated_text']
     return corrected_text