Update app.py

app.py

@@ -169,16 +169,16 @@ os.system("rm -R -- */")
 with col:
     with st.container():
         # Create heading and description
-        st.markdown("<h1 align='center'>Cavity Detection Tool</h1>", unsafe_allow_html=True)    
+        st.markdown("<h1 align='center'>Cavity Detection Tool (<a href='https://github.com/tomasplsek/CADET'>CADET</a>)</h1>", unsafe_allow_html=True)    
         # 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.")
         # st.markdown("To use this tool: upload your image, select the scale of interest, make a prediction, and decompose it into individual cavities!")
         # st.markdown("Input images should be FITS files in units of counts, centred at the galaxy center, and point sources should be filled with surrounding background ([dmfilth](https://cxc.cfa.harvard.edu/ciao/ahelp/dmfilth.html)).")
         # st.markdown("If you use this tool for your research, please cite [Plšek et al. 2023](https://arxiv.org/abs/2304.05457)")
 
         st.markdown("<div style='border-radius:5px;background-color:#F3F4F6;padding-top:8px;padding-bottom:8px;padding-left:14px;padding-right:14px;line-height:140%;font-size:120%'>\
-Cavity Detection Tool (CADET) is a machine learning pipeline trained to detect <b>X-ray cavities</b> from <b><em>Chandra</em></b> images of early-type galaxies, groups, and clusters. \
-To use this tool:<br><b>1)</b> upload your image<br><b>2)</b> select the scale of interest<br><b>3)</b> make a prediction<br><b>4)</b> decompose into individual cavities\
-<br>If you use this tool for your research, please cite <a href='https://arxiv.org/abs/2304.05457'>Plšek et al. 2023</a>.\
+Cavity Detection Tool (CADET) is a machine learning pipeline trained to detect X-ray cavities from <em>Chandra</em> images of early-type galaxies, groups, and clusters.\
+<br>To use this tool:<br><b>1)</b> upload your FITS file<br><b>2)</b> select the scale of interest<br><b>3)</b> make a prediction<br><b>4)</b> decompose into individual cavities\
+<br style='margin-bottom: 6px'>If you use this tool for your research, please cite <a href='https://arxiv.org/abs/2304.05457'>Plšek et al. 2023</a>.\
 </div><br>", unsafe_allow_html=True)
 
 # Input images should be FITS files in units of counts, centred at the galaxy center, and point sources should be filled with surrounding background \