Update app.py

app.py

@@ -44,6 +44,15 @@ def correct_grammar_with_languagetool(text):
     corrected_text = language_tool_python.utils.correct(text, matches)
     return corrected_text
 
+# AI detection function using DistilBERT
+def detect_ai_generated(text):
+    inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512).to(device)
+    with torch.no_grad():
+        outputs = model(**inputs)
+        probabilities = torch.softmax(outputs.logits, dim=1)
+    ai_probability = probabilities[0][1].item()  # Probability of being AI-generated
+    return f"AI-Generated Content Probability: {ai_probability:.2f}%"
+
 # Function to get synonyms using NLTK WordNet
 def get_synonyms_nltk(word, pos):
     synsets = wordnet.synsets(word, pos=pos)