Update app.py

app.py

@@ -65,6 +65,5 @@ else:
     st.write(cnn.apply({"params": params}, input).argmax(axis=1)[0])
 
 with st.expander("See first convolutional layer"):
-    rgb = np.repeat(np.expand_dims(np.array(params["Conv_0"]["kernel"].squeeze()[:,:,0]+1)*127., axis=0), repeats=3, axis=0)
-    print(rgb)
-    st.image(Image.fromarray(rgb, mode="RGB").transpose(1,2,0).resize((28, 28)))
+    rgb = np.repeat(np.expand_dims(np.array(params["Conv_0"]["kernel"].squeeze()[:,:,0]+1)*127., axis=0), repeats=3, axis=0).transpose(1,2,0)
+    st.image(Image.fromarray(rgb, mode="RGB").resize((28, 28)))