Update app.py

app.py

@@ -96,9 +96,10 @@ def main():
                 prediction_text = "No Tuberculosis is found based on the audio data and clinical information."
                 probability_label = "Negative Probability"
 
-            st.write('**Prediction:**')
-            st.write(f"{prediction_text}")
-            st.write(f"{probability_label}: {result}")
+            # Display prediction in a popup
+            st.markdown(f"<h2 style='text-align:center;'>Prediction:</h2>", unsafe_allow_html=True)
+            st.markdown(f"<h3 style='text-align:center; font-weight:bold;'>{prediction_text}</h3>", unsafe_allow_html=True)
+            st.markdown(f"<h4 style='text-align:center;'>{probability_label}: {result}</h4>", unsafe_allow_html=True)
 
 if __name__ == '__main__':
     main()