sashtech commited on
Commit
5bfbf54
·
verified ·
1 Parent(s): aed9390

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -6,7 +6,7 @@ import subprocess
6
  import nltk
7
  from nltk.corpus import wordnet
8
  from gensim import downloader as api
9
- from gingerit.gingerit import GingerIt
10
 
11
  # Ensure necessary NLTK data is downloaded
12
  nltk.download('wordnet')
@@ -29,9 +29,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
29
  tokenizer_ai = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
30
  model_ai = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english").to(device)
31
 
32
- # Initialize GingerIt parser
33
- parser = GingerIt()
34
-
35
  # AI detection function using DistilBERT
36
  def detect_ai_generated(text):
37
  inputs = tokenizer_ai(text, return_tensors="pt", truncation=True, max_length=512).to(device)
@@ -100,10 +97,10 @@ def paraphrase_with_spacy_nltk(text):
100
 
101
  return corrected_text
102
 
103
- # Function to correct grammar using GingerIt
104
  def correct_grammar(text):
105
- result = parser.parse(text)
106
- return result['result']
107
 
108
  # Combined function: Paraphrase -> Capitalization -> Grammar Correction
109
  def paraphrase_and_correct(text):
 
6
  import nltk
7
  from nltk.corpus import wordnet
8
  from gensim import downloader as api
9
+ from textblob import TextBlob
10
 
11
  # Ensure necessary NLTK data is downloaded
12
  nltk.download('wordnet')
 
29
  tokenizer_ai = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
30
  model_ai = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english").to(device)
31
 
 
 
 
32
  # AI detection function using DistilBERT
33
  def detect_ai_generated(text):
34
  inputs = tokenizer_ai(text, return_tensors="pt", truncation=True, max_length=512).to(device)
 
97
 
98
  return corrected_text
99
 
100
+ # Function to correct grammar using TextBlob
101
  def correct_grammar(text):
102
+ blob = TextBlob(text)
103
+ return str(blob.correct())
104
 
105
  # Combined function: Paraphrase -> Capitalization -> Grammar Correction
106
  def paraphrase_and_correct(text):