deepsaif / utils /visualization.py
22GC22's picture
Upload 17 files
2fbad05 verified
import pandas as pd
import matplotlib.pyplot as plt
import streamlit as st
def display_results(results):
st.write("### Detection Results")
df = pd.DataFrame(results.items(), columns=["Model", "Probability (%)"])
st.table(df)
st.write("### Visualization")
fig, ax = plt.subplots()
df.plot.bar(x="Model", y="Probability (%)", ax=ax, legend=False)
ax.set_ylabel("Probability (%)")
st.pyplot(fig)