Update app.py

app.py

@@ -40,16 +40,11 @@ else:
     predictions, time_take = import_and_predict(image, model)
     st.write("Time taken to predict is ", time_take, "second")
     st.write(predictions)
-    print(predictions)
+    print(predictions.shape)
     
     print(np.argmax(predictions))
     st.write(np.argmax(predictions))
-    
-    #st.write('each member of pred')
-    #st.write(predictions.shape)
-    #st.write(predictions[0])
-    #st.write(predictions[1])
-    
+        
     if predictions[0][0] < 0.5:
         st.write("This is SFW image :sunglasses:")
     else: