Update app.py

app.py

@@ -273,7 +273,7 @@ if st.button('Submit'):
 
     # Use the second column to display the label aligned to the right with a border
     with col2:
-        st.markdown(f"<div style='text-align: right; padding-right: 16px; border: 2px solid #f0f2f6; padding: 4px; border-radius: 5px; margin: 10px 0;'><strong>{predicted_label}</strong></div>", unsafe_allow_html=True)
+        st.markdown(f"<div style='text-align: right; padding-right: 16px; border: 2px solid #f0f2f6; padding: 4px; border-radius: 5px; margin: 10px 0;'><strong>{random_label}</strong></div>", unsafe_allow_html=True)
 
 
     # Add prediction to the DataFrame