Update app.py

app.py

@@ -29,14 +29,16 @@ def plot_image(image_array, scale):
     plt.gca().add_patch(Rectangle((x0, x0), scale*128, scale*128, linewidth=1, edgecolor='w', facecolor='none'))
     
     plt.axis('off')
-    with col1: st.pyplot()
+    # with col1: 
+    st.pyplot()
 
 # Define function to plot the prediction
 def plot_prediction(pred):
     plt.figure(figsize=(4, 4))
     plt.imshow(pred, origin="lower")
     plt.axis('off')
-    with col2: st.pyplot()
+    # with col2: 
+    st.pyplot()
 
 def cut(data0, wcs0, scale=1):
     shape = data0.shape[0]
@@ -67,12 +69,12 @@ if uploaded_file is not None:
         data = hdul[0].data
         wcs = WCS(hdul[0].header)
 
-        max_scale = data // 128
         # Add a slider to change the scale
         with col1:
+            max_scale = int(data // 128)
             scale = st.slider("Scale", 1, max_scale, 1, 1)
 
-        plot_image(np.log10(data+1), scale)
+            plot_image(np.log10(data+1), scale)
 
         with col2:
             if st.button('Detect cavities'):