zamborg commited on
Commit
2fa0369
1 Parent(s): 07d062e

potentially breaking advanced options

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -97,10 +97,11 @@ cap_prompt = st.sidebar.text_input(
97
 
98
  _ = st.sidebar.button("Regenerate Caption")
99
 
100
- # advanced = st.sidebar.checkbox("Advanced Options")
101
-
102
- # if advanced:
103
- # nuc_size = st.sidebar.slider("")
 
104
 
105
  if uploaded_image is None and submitted:
106
  st.write("Please select a file to upload")
@@ -128,7 +129,8 @@ else:
128
  show = st.image(show_image)
129
  show.image(show_image, "Your Image")
130
 
131
- gen_show_caption(sub, imageLoader.text_transform(cap_prompt))
 
132
 
133
  # from model import *
134
  # sample_images = get_samples()
97
 
98
  _ = st.sidebar.button("Regenerate Caption")
99
 
100
+ advanced = st.sidebar.checkbox("Advanced Options")
101
+ num_captions=1
102
+ if advanced:
103
+ nuc_size = st.sidebar.slider("Nucelus Size:", min_value=0.0, max_value=1.0, value=0.8, step=0.05)
104
+ num_captions = st.sidebar.slider("Number of Captions to Predict", options=[1,2,3,4,5], value=1)
105
 
106
  if uploaded_image is None and submitted:
107
  st.write("Please select a file to upload")
129
  show = st.image(show_image)
130
  show.image(show_image, "Your Image")
131
 
132
+ for i in range(num_captions):
133
+ gen_show_caption(sub, imageLoader.text_transform(cap_prompt))
134
 
135
  # from model import *
136
  # sample_images = get_samples()