Update app.py

app.py

@@ -198,7 +198,10 @@ def main():
                     print(f"This is likely an unwatermarked audio: {result2}")
                     st.markdown("Probability of audio being watermarked: ")
                     st.markdown(result2)
-                    st.markdown("This is likely an unwatermarked audio!")
+                    if result2 < 0.5:
+                        st.markdown("This is likely an unwatermarked audio!")
+                    else:
+                        st.markdown("This is likely an watermarked audio!")
 
     except RuntimeError:
         st.error("Please input audio with one channel (mono-channel)")