testing caption_prompting

app.py

@@ -5,7 +5,7 @@ import time
 import json
 sys.path.append("./virtex/")
 
-def gen_show_caption(sub_prompt=None):
+def gen_show_caption(sub_prompt=None, cap_prompt = None):
     with st.spinner("Generating Caption"):
         subreddit, caption = virtexModel.predict(image_dict, sub_prompt=sub_prompt)
         st.header("Predicted Caption:\n\n")
@@ -35,6 +35,21 @@ sample_image = st.sidebar.selectbox(
     sample_images
 )
 
+st.sidebar.title("Select a Subreddit")
+sub = st.sidebar.selectbox(
+    "Select None for a Predicted Subreddit",
+    valid_subs
+)
+
+st.sidebar.title("Write a Custom Prompt")
+cap_prompt = st.text_input(
+    "Leave this blank for an unbiased caption", 
+    value="This is a photo of"
+)
+st.write("=====================")
+st.write(cap_prompt)
+st.write("=====================")
+
 if st.sidebar.button("Random Sample Image"):
     random_image = get_rand_img(sample_images)
     sample_image = None
@@ -46,11 +61,6 @@ with st.sidebar.form("file-uploader-form", clear_on_submit=True):
     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:
@@ -68,7 +78,7 @@ else:
     show = st.image(image)
     show.image(image, "Your Image")
     
-    gen_show_caption(sub)
+    gen_show_caption(sub, cap_prompt)
         
     image.close()