Update app.py

app.py

@@ -29,7 +29,7 @@ from huggingface_hub import from_pretrained_keras
 def plot_image(image, scale):
     plt.figure(figsize=(4, 4))
     x0 = image.shape[0] // 2 - scale * 128 / 2
-    plt.imshow(image, origin="lower", norm=LogNorm())
+    plt.imshow(np.log10(image+1), origin="lower") #, norm=LogNorm())
     plt.gca().add_patch(Rectangle((x0-0.5, x0-0.5), scale*128, scale*128, linewidth=1, edgecolor='w', facecolor='none'))
     plt.axis('off')
     plt.tight_layout()
@@ -195,7 +195,6 @@ with col_1:
 with col_2:
     st.markdown("<br style='margin:15px 0'>", unsafe_allow_html=True)
     example = st.button("Example", key="example")
-    st.markdown('<style>div.row-widget.stButton > div{margin-left: auto;}</style>', unsafe_allow_html=True)
     
     # with col_2:
     #     st.markdown("### Examples")
@@ -226,7 +225,7 @@ if st.session_state.get("example", True):
     data, wcs = load_file(f"{fname}.fits")
 
 if "data" not in locals():
-    data = np.zeros((128,128))
+    data = np.ones((128,128)) * (-1)
         
 # Make six columns for buttons
 _, col1, col2, col3, col4, col5, col6, _ = st.columns([bordersize,0.5,0.5,0.5,0.5,0.5,0.5,bordersize])