app.py updates

app.py

@@ -59,7 +59,7 @@ col1, col2 = st.columns(2)
 
 with col1:
 
-  current_image=ds[st.session_state["img"]]['image']
+  current_image=np.array(ds[st.session_state["img"]]['image'])[::2,::2,:]
   draw = ImageDraw.Draw(current_image)
 
   if st.session_state["draw"]:
@@ -91,6 +91,6 @@ with col1:
 with col2:
 
   for i in range(4):
-    st.image(np.array(st.session_state["sideimg"][i])[::4,::4,:])
+    st.image(np.array(ds[st.session_state["sideimg"][i]]['image'])[::4,::4,:])