Update app.py

app.py

@@ -35,7 +35,17 @@ def detect_ai_generated(text):
         outputs = model_ai(**inputs)
         probabilities = torch.softmax(outputs.logits, dim=1)
     ai_probability = probabilities[0][1].item() * 100  # Probability of being AI-generated
-    return f"AI-Generated Content Probability: {ai_probability:.2f}%"
+    human_probability = 100 - ai_probability  # Probability of being Human-written
+    
+    # Determine the label based on the higher probability
+    if ai_probability > human_probability:
+        label = "AI"
+        probability = ai_probability
+    else:
+        label = "Human"
+        probability = human_probability
+    
+    return f"The content is {probability:.2f}% {label} Written"
 
 # Function to get synonyms using NLTK WordNet
 def get_synonyms_nltk(word, pos):