Update app.py

app.py

@@ -20,11 +20,39 @@ def fill_mask(sentences):
 def replace_mask(sentence, predicted_word):
     return sentence.replace("<mask>", f"**{predicted_word}**")
 
-st.title("Fill Mask | Zabantu-nso-120m")
+st.write(f"")
+img1, img2, img3 = st.columns(3)
+with img2:
+    with st.container(border=False):
+        st.image("logo_transparent_small.png")
+
+st.markdown("""
+    <div style='text-align: center;'>
+        <a href='https://github.com/dsfsi' target='_blank'>Github</a> |
+        <a href='https://docs.google.com/forms/d/e/1FAIpQLSf7S36dyAUPx2egmXbFpnTBuzoRulhL5Elu-N1eoMhaO7v10w/viewform' target='_blank'>Feedback Form</a> |
+        <a href='https://huggingface.co/papers/1911.02116' target='_blank'>arxiv</a>
+    </div>
+""", unsafe_allow_html=True)
+ 
+st.markdown("""
+    <div style='text-align: center;'>
+      <h2>Fill Mask | Zabantu-nso-120m</h2>    
+    </div>
+""", unsafe_allow_html=True)
+st.write(f"")
 st.write(f"")
 
 st.markdown("This is a variant of Zabantu pre-trained on a monolingual dataset of Sepedi(nso) sentences on a transformer network with 120 million traininable parameters.")
 
+with st.expander("More information about the space"):
+    st.write('''
+        Authors: Alexis Conneau, Kartikay Khandelwal, Naman Goyal, Vishrav Chaudhary, Guillaume Wenzek, Francisco Guzmán, Edouard Grave, Myle Ott, Luke Zettlemoyer, Veselin Stoyanov
+    ''')
+    cit1,cit2 = st.columns(2)
+    # with cit1:
+    # with cit2:
+
+
 col1, col2 = st.columns(2)
 
 if 'text_input' not in st.session_state: