Update app.py

app.py

@@ -320,17 +320,16 @@ if press_me_button:
     # extra_trees_name =  author_map.get(extra_trees_prediction[0], extra_trees_prediction[0])
 
     cnn_name, ridge_name, extra_trees_name = get_author_display_name(predicted_author, ridge_prediction, extra_trees_prediction)
-    
+    with st.expander("Show More Details..."):
+        st.write(f"""
+        Ridge: {ridge_name}
+        ExtraTree: {extra_trees_name}
+        CNN: {cnn_name}
+        """)
     if ridge_prediction == extra_trees_prediction == predicted_author:
         st.success(f"Most likely written by: **{ridge_name}**", icon="✅")
         st.info("We are quite confident in the accuracy of this result.", icon="ℹ️")
-        # rain(
-        #     emoji="😃",
-        #     font_size=54,
-        #     falling_speed=5,
-        #     animation_length="infinite",
-        # )
-
+        
     else:
         # Repeat the text with a space at the end of each iteration