Spaces:

jayasuriyaK
/

nsfw_classifier

Sleeping

jayasuriyaK commited on May 11

Commit

70901e5

•

1 Parent(s): 2e020c0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,7 +115,7 @@ if uploaded_file is not None:
     outputs = model(**inputs)
     predictions = outputs.logits.softmax(dim=-1)
     print(predictions[0][0],predictions[0][1])
-    if predictions[0][1]>predictions[0][0]:
         print('safe')
         st.write('Safe for Work')
     else:

     outputs = model(**inputs)
     predictions = outputs.logits.softmax(dim=-1)
     print(predictions[0][0],predictions[0][1])
+    if predictions[0][0]>predictions[0][1]:
         print('safe')
         st.write('Safe for Work')
     else: