Spaces:

PrakhAI
/

HandwrittenDigit

Sleeping

PrakhAI commited on Jul 3, 2023

Commit

52b6792

1 Parent(s): 230339a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,4 +65,4 @@ else:
     st.write(cnn.apply({"params": params}, input).argmax(axis=1)[0])
 with st.expander("See first convolutional layer"):
-    st.image(Image.fromarray((np.array(params["Conv_0"]["kernel"].squeeze()[:,:,0])+1)*127.))

     st.write(cnn.apply({"params": params}, input).argmax(axis=1)[0])
 with st.expander("See first convolutional layer"):
+    st.image(Image.fromarray((np.array(params["Conv_0"]["kernel"].squeeze()[:,:,0])+1)*127., mode="I"))