RishabhBhardwaj commited on
Commit
ccd23b6
1 Parent(s): db1e7ef
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -46,13 +46,15 @@ if st.button("Evaluate"):
46
  st.subheader("Evaluation Result:")
47
  st.write(f"The text is evaluated as: **{prediction.upper()}**")
48
 
49
- #st.subheader("Model Output:")
50
- #st.write(output_decoded)
51
  else:
52
  st.warning("Please enter some text to evaluate.")
53
 
54
- # Add some information about the model
55
- #st.sidebar.header("About")
56
- #st.sidebar.info("This app uses the WalledGuard-C model to evaluate the safety of input text. It determines whether the text is asking for or containing unsafe information.")
 
57
 
58
- #gr.load("models/walledai/walledguard-c").launch()
 
 
 
 
46
  st.subheader("Evaluation Result:")
47
  st.write(f"The text is evaluated as: **{prediction.upper()}**")
48
 
 
 
49
  else:
50
  st.warning("Please enter some text to evaluate.")
51
 
52
+ # Add information about Walled Guard Advanced
53
+ st.markdown("---")
54
+ st.write("For a more performant version, check out Walled Guard Advanced.")
55
+ st.write("Connect with us at admin@walled.ai for more information.")
56
 
57
+ # Add logo at the bottom center
58
+ col1, col2, col3 = st.columns([1,2,1])
59
+ with col2:
60
+ st.image("logo.png", use_column_width=True)