added center column context manager

app.py

@@ -32,8 +32,10 @@ def gen_show_caption(sub_prompt=None, cap_prompt = ""):
             """, 
             unsafe_allow_html=True)
     
+_, center, _ = st.columns(3)
 
-st.title("Image Captioning Demo from RedCaps")
+with center:
+    st.title("Image Captioning Demo from RedCaps")
 st.sidebar.markdown(
     """
     ### Image Captioning Model from VirTex trained on RedCaps
@@ -50,12 +52,6 @@ with st.spinner("Loading Model"):
     virtexModel, imageLoader, sample_images, valid_subs = create_objects()
     
 
-# staggered = st.sidebar.checkbox("Iteratively Generate Captions")
-
-# if staggered:
-#     pass
-# else:
-
 select_idx = None
 
 st.sidebar.title("Select a sample image")
@@ -129,10 +125,10 @@ image_dict = imageLoader.transform(image)
 show_image = imageLoader.show_resize(image)
 
 show = st.image(show_image)
-show.image(show_image, "Your Image")
-
+show.image(show_image)
 
-if sub is None and imageLoader.text_transform(cap_prompt) is not "":
-    st.write("Without a specified subreddit we default to /r/pics")
-for i in range(num_captions):
-    gen_show_caption(sub, imageLoader.text_transform(cap_prompt))
+with center:
+    if sub is None and imageLoader.text_transform(cap_prompt) is not "":
+        st.write("Without a specified subreddit we default to /r/pics")
+    for i in range(num_captions):
+        gen_show_caption(sub, imageLoader.text_transform(cap_prompt))