Spaces:

PrakhAI
/

HandwrittenDigit

Sleeping

PrakhAI commited on Jul 3, 2023

Commit

d92da68

1 Parent(s): e8bcaf6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -67,4 +67,4 @@ else:
 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").resize((28, 28)))

 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)))