Update app.py

app.py

@@ -316,6 +316,24 @@ if press_me_button:
         "human": "Human-Written"
     }
 
+
+    max_cnn_prob_name = sorted_probabilities[0][0]
+    max_cnn_prob = float(sorted_probabilities[0][1])
+
+    def classify_human_or_ai(ridge_prob, extra_trees_prob, cnn_prob):
+        threshold = 0.5  # You can adjust this threshold as needed
+        if max(ridge_prob, extra_trees_prob, cnn_prob) < threshold:
+            return "Human-Written"
+        else:
+            return "AI Generated"
+
+    # Determine the classification
+    classification = classify_human_or_ai(max_cnn_prob, max(ridge_prediction), max(extra_trees_prediction))
+
+    # Display the classification result
+    with st.expander("Classification Result"):
+        st.write(f"The input text is classified as: **{classification}**")
+
     cnn_name, ridge_name, extra_trees_name = get_author_display_name(predicted_author, ridge_prediction, extra_trees_prediction)
     with st.expander("Prediction Details (Click Here)..."):
         st.write(f"Ridge: {ridge_name}")
@@ -328,8 +346,7 @@ if press_me_button:
             st.write(f"{display_name}: {prob * 100:.2f}%")
             st.progress(float(prob))
             
-    max_cnn_prob_name = sorted_probabilities[0][0]
-    max_cnn_prob = float(sorted_probabilities[0][1])
+    
 
     if word_count < 10.0 or word_count > 1081.0:
         st.warning("For better prediction input texts between 10 and 1081", icon="ℹ️")
@@ -376,7 +393,6 @@ if press_me_button:
             st.success(f"2- **{cnn_name}**", icon="✅")
             st.success(f"3- **{extra_trees_name}**", icon="✅")
 
-
  
 
 # Using expander to make FAQ sections
@@ -433,6 +449,3 @@ with st.expander("Can I use this as evidence?"):
     """)
 
 
-
-
-