Hugging Face's logo

Spaces:
sashtech
/
huamnifierWithSimpleGrammer
Running

App Files
sashtech commited on
Commit
5b1fe73
·
verified ·
1 Parent(s): c3f5d2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ tokenizer_ai = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-
29
  model_ai = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english").to(device)
30
 
31
  # Load Grammar Correction model (T5) from Hugging Face
32
- grammar_corrector = pipeline('text2text-generation', model='prithivida/grammar-error-correcter')
33
 
34
  # AI detection function using DistilBERT
35
  def detect_ai_generated(text):
 
29
  model_ai = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english").to(device)
30
 
31
  # Load Grammar Correction model (T5) from Hugging Face
32
+ grammar_corrector = pipeline('text2text-generation', model='vennify/t5-base-grammar-correction')
33
 
34
  # AI detection function using DistilBERT
35
  def detect_ai_generated(text):