Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -129,15 +129,3 @@ prediction, targets = inference()
|
|
129 |
st.write("before argmax")
|
130 |
st.write(prediction)
|
131 |
st.write(targets)
|
132 |
-
|
133 |
-
st.write("after argmax axis 1")
|
134 |
-
targets = np.argmax(targets, axis=1)
|
135 |
-
prediction = np.argmax(prediction, axis=1)
|
136 |
-
st.write(prediction)
|
137 |
-
st.write(targets)
|
138 |
-
|
139 |
-
st.write("after argmax axis 2")
|
140 |
-
targets = np.argmax(targets, axis=2)
|
141 |
-
prediction = np.argmax(prediction, axis=2)
|
142 |
-
st.write(prediction)
|
143 |
-
st.write(targets)
|
|
|
129 |
st.write("before argmax")
|
130 |
st.write(prediction)
|
131 |
st.write(targets)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|