Spaces:

jayasuriyaK
/

nsfw_classifier

Sleeping

jayasuriyaK commited on Apr 30

Commit

8a9ee62

•

1 Parent(s): ebb319e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -105,7 +105,7 @@ 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 for Work')
     else:
         print('Not safe')
-        st.write('Not Safe for Work')

     print(predictions[0][0],predictions[0][1])
     if predictions[0][0]>predictions[0][1]:
         print('safe')
+        st.success('Safe for Work')
     else:
         print('Not safe')
+        st.success('Not Safe for Work')