Update app.py

app.py

@@ -322,14 +322,15 @@ if press_me_button:
 
     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:
+        # Convert to float before comparison
+        if max(float(ridge_prob), float(extra_trees_prob), float(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}**")
@@ -351,6 +352,10 @@ if press_me_button:
     if word_count < 10.0 or word_count > 1081.0:
         st.warning("For better prediction input texts between 10 and 1081", icon="ℹ️")
 
+    elif classification == "Human-Written":
+        st.success("The input text is classified as **Human-Written**.", icon="✅")
+        st.info("We are quite confident in the accuracy of this result.", icon="ℹ️")
+        
     elif word_count < 256: 
         if ridge_prediction == extra_trees_prediction == predicted_author:
             st.success(f"Most likely written by: **{cnn_name}**", icon="✅")