Update app.py

app.py

@@ -11,8 +11,7 @@ st.set_option('deprecation.showPyplotGlobalUse', False)
 
 st.title("Cavity Detection Tool")
 
-st.text("Cavity Detection Tool (CADET) is a machine learning pipeline trained to detect X-ray cavities from noisy Chandra images of early-type galaxies.\
-To use this tool upload your image, select the scale of interest and make a prediction! If you use output of this tool in your research please cite [Plšek et al. 2023](https://arxiv.org/abs/2304.05457)")
+st.text("Cavity Detection Tool (CADET) is a machine learning pipeline trained to detect\n X-ray cavities from noisy Chandra images of early-type galaxies.To use this tool upload\n your image, select the scale of interest and make a prediction! If you use output of this\n tool in your research please cite [Plšek et al. 2023](https://arxiv.org/abs/2304.05457)")
 
 model = load_model("CADET.hdf5")
 
@@ -31,7 +30,8 @@ def plot_image(image_array, scale):
     
     plt.axis('off')
     fig.set_size_inches((4,4))
-    st.pyplot()
+    with col1:
+        st.pyplot()
 
 # Define function to plot the prediction
 def plot_prediction(image_array, pred):
@@ -43,7 +43,8 @@ def plot_prediction(image_array, pred):
     # plt.subplot(1, 2, 2)
     plt.imshow(pred, origin="lower")
     plt.axis('off')
-    col2.pyplot()
+    with col2:
+        st.pyplot()
 
 def cut(data0, wcs0, scale=1):
     shape = data0.shape[0]