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):
|