Update app.py

app.py

@@ -11,7 +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\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)")
+st.markdown("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)")
 
 model = load_model("CADET.hdf5")
 
@@ -20,6 +20,9 @@ uploaded_file = st.file_uploader("Choose a FITS file", type=['fits'])
 
 col1, col2 = st.columns(2)
 
+col1.subheader("Input image")
+col.subheader("CADET prediction")
+
 # Define function to plot the uploaded image
 def plot_image(image_array, scale):
     plt.figure(figsize=(4, 4))