Update app.py

app.py

@@ -16,15 +16,12 @@ def gen_show_caption(sub_prompt=None, cap_prompt=""):
         st.markdown(
             f"""
             <style>
-                red{{
-                    color:#c62828
-                }}
-                blue{{
-                    color:#2a72d5
-                }}
+                red {{ color:#c62828; font-size: 1.5rem }}
+                blue {{ color:#2a72d5; font-size: 1.5rem }}
+                remaining {{ color: black; font-size: 1.5rem }}
             </style>
 
-            #### <red> r/{subreddit} </red> <blue> {cap_prompt} </blue> {caption}
+            - <red> r/{subreddit} </red> <blue> {cap_prompt} </blue><remaining> {caption} </remaining>
             """,
             unsafe_allow_html=True,
         )
@@ -65,7 +62,7 @@ cap_prompt = st.sidebar.text_input("Write the start of your caption below", valu
 _ = st.sidebar.button("Regenerate Caption")
 
 
-st.sidebar.title("Advanced Options:")
+st.sidebar.title("Advanced Options")
 num_captions = st.sidebar.select_slider(
     "Number of Captions to Predict", options=[1, 2, 3, 4, 5], value=1
 )
@@ -76,15 +73,6 @@ nuc_size = st.sidebar.slider(
     value=0.8,
     step=0.05,
 )
-st.sidebar.markdown(
-    """
-*Please note that this model was explicitly not trained on images of people, and as a result is not designed to caption images with humans.
-
-This demo accompanies our paper RedCaps.
-
-Created by Karan Desai, Gaurav Kaul, Zubin Aysola, Justin Johnson
-    """
-)
 # ----------------------------------------------------------------------------
 
 virtexModel.model.decoder.nucleus_size = nuc_size
@@ -113,8 +101,12 @@ st.markdown("""
 Caption your own images or try out some of our sample images.
 You can also generate captions as if they are from specific subreddits,
 as if they start with a particular prompt, or even both.
-
 Tweet your results with `#redcaps`!
+
+**Note:** This model was not trained on images of people,
+hence may not generate accurate captions describing humans.
+For more details, visit [redcaps.xyz](https://redcaps.xyz) check out
+our [NeurIPS 2021 paper](https://openreview.net/forum?id=VjJxBi1p9zh).
 """)
 
 _, center, _ = st.columns([1, 18, 1])