Spaces:
Runtime error
Runtime error
potentially breaking advanced options
Browse files
app.py
CHANGED
|
@@ -97,10 +97,11 @@ cap_prompt = st.sidebar.text_input(
|
|
| 97 |
|
| 98 |
_ = st.sidebar.button("Regenerate Caption")
|
| 99 |
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
|
|
|
| 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 |
-
|
|
|
|
| 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()
|