Spaces:
Running
Running
Update app.py
Browse files
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
|
| 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
|
| 104 |
def correct_grammar(text):
|
| 105 |
-
|
| 106 |
-
return
|
| 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):
|