Update app.py

app.py

@@ -50,7 +50,7 @@ def get_distance(predictions):
 
     return detections
 
-def distinguish_rows(lst, thresh=15):
+def distinguish_rows(lst, thresh=10):
     """Function to help distinguish unique rows"""
     sublists = []
     for i in range(0, len(lst)-1):
@@ -64,7 +64,7 @@ def distinguish_rows(lst, thresh=15):
     yield sublists
 
 # Title of the app
-st.title("Image Input App")
+st.title("NSFW Content Detector")
 
 # File uploader widget
 uploaded_file = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
@@ -105,7 +105,9 @@ if uploaded_file is not None:
     print(predictions[0][0],predictions[0][1])
     if predictions[0][0]>predictions[0][1]:
         print('safe')
-        st.write('safe')
+        st.write('Safe for Work')
+        st.text_area("The given content has", "SFW")
     else:
         print('Not safe')
-        st.write('n safe')
+        st.write('Not Safe for Work')
+        st.text_area("The given content has", "NSFW")