added sub conditioning

app.py

@@ -2,8 +2,16 @@ import streamlit as st
 import io
 import sys
 import time
+import json
 sys.path.append("./virtex/")
 
+def gen_show_caption():
+    with st.spinner("Generating Caption"):
+        subreddit, caption = virtexModel.predict(image_dict)
+        st.header("Predicted Caption:\n\n")
+        st.subheader(f"r/{subreddit}:\t{caption}\n")
+    
+
 st.title("Image Captioning Demo from Redcaps")
 st.sidebar.markdown(
     """
@@ -12,15 +20,12 @@ st.sidebar.markdown(
 )
 
 with st.spinner("Loading Model"):
-    st.write("DEBUG PRINTING ==========")
-    start = time.time()
     from model import *
-    st.write(f"Import TIME: {time.time()-start}")
     sample_images = get_samples()
-    start = time.time()
     virtexModel = VirTexModel()
     imageLoader = ImageLoader()
-    st.write(f"model TIME: {time.time()-start}")
+    valid_subs = json.load(open(VALID_SUBREDDITS_PATH))
+    valid_subs.insert(0, None)
 
 random_image = get_rand_img(sample_images)
 
@@ -40,6 +45,13 @@ with st.sidebar.form("file-uploader-form", clear_on_submit=True):
     submitted = st.form_submit_button("Submit")
     if uploaded_file is not None and submitted:
         uploaded_image = Image.open(io.BytesIO(uploaded_file.getvalue()))
+        
+st.sidebar.title("Select a Subreddit")
+sub = st.sidebar.selectbox(
+    "Select None for a Predicted Subreddit",
+    valid_subs
+)
+
 
 if uploaded_image is None and submitted:
     st.write("Please select a file to upload")
@@ -56,10 +68,7 @@ else:
     show = st.image(image)
     show.image(image, "Your Image")
     
-    with st.spinner("Generating Caption"):
-        subreddit, caption = virtexModel.predict(image_dict)
-        st.header("Predicted Caption:\n\n")
-        st.subheader(f"r/{subreddit}:\t{caption}\n")
+    gen_show_caption()
         
     image.close()