Update app.py

app.py

@@ -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]:
+    if predictions[0][0]>predictions[0][1]:
         print('safe')
         st.write('Safe for Work')
     else: