Update app.py

app.py

@@ -41,9 +41,6 @@ else:
     st.write("Time taken to predict is ", time_take, "second")
     st.write(predictions)
     print(predictions.shape)
-    
-    print(np.argmax(predictions))
-    st.write(np.argmax(predictions))
         
     if predictions[0][0] < 0.5:
         st.write("This is SFW image :sunglasses:")