Update app.py

app.py

@@ -328,6 +328,8 @@ if press_me_button:
             st.progress(float(prob))
             
     st.write(f"Sorted probab: {sorted_probabilities[0][1] }")
+
+    max_cnn_prob_name = sorted_probabilities[0][0]
     max_cnn_prob = float(sorted_probabilities[0][1])
 
     
@@ -413,6 +415,9 @@ if press_me_button:
         predicted_author, author_probabilities = predict_author(new_text, loaded_model, tokenizer, label_encoder)
         sorted_probabilities = sorted(author_probabilities.items(), key=lambda x: x[1], reverse=True)
 
+        new_max_cnn_prob_name = sorted_probabilities[0][0]
+        new_max_cnn_prob = float(sorted_probabilities[0][1])
+        
         # Get disply name
         cnn_name, ridge_name, extra_trees_name = get_author_display_name(predicted_author, ridge_prediction, extra_trees_prediction)
         with st.expander("2nd iteration Details..."):
@@ -434,6 +439,16 @@ if press_me_button:
             #     falling_speed=5,
             #     animation_length="infinite",
             # )
+        elif new_max_cnn_prob == max_cnn_prob and new_max_cnn_prob_name == max_cnn_prob_name:
+            st.success(f"Most likely written by: **{cnn_name}**", icon="✅")
+            st.warning(f"**Notice:** The input text has been magnified {amplify} times to better capture its characteristics and patterns.", icon="⚠️")
+            st.write("_" * 30)
+            # rain(
+            #     emoji="😐",
+            #     font_size=54,
+            #     falling_speed=5,
+            #     animation_length="infinite",
+            # )
             
         elif ridge_prediction == extra_trees_prediction:
             st.success(f"Most likely written by: **{ridge_name}**", icon="✅")