Update app.py

app.py

@@ -129,15 +129,3 @@ prediction, targets = inference()
 st.write("before argmax")
 st.write(prediction)
 st.write(targets)
-
-st.write("after argmax axis 1")
-targets = np.argmax(targets, axis=1)
-prediction = np.argmax(prediction, axis=1)
-st.write(prediction)
-st.write(targets)
-
-st.write("after argmax axis 2")
-targets = np.argmax(targets, axis=2)
-prediction = np.argmax(prediction, axis=2)
-st.write(prediction)
-st.write(targets)