Spaces:
Build error
Build error
advanced options always on
Browse files
app.py
CHANGED
@@ -103,12 +103,11 @@ cap_prompt = st.sidebar.text_input(
|
|
103 |
|
104 |
_ = st.sidebar.button("Regenerate Caption")
|
105 |
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
-
virtexModel.model.decoder.nucleus_size = nuc_size
|
112 |
|
113 |
image_file = sample_image
|
114 |
|
|
|
103 |
|
104 |
_ = st.sidebar.button("Regenerate Caption")
|
105 |
|
106 |
+
|
107 |
+
st.sidebar.write("Advanced Options:")
|
108 |
+
num_captions = st.sidebar.select_slider("Number of Captions to Predict", options=[1,2,3,4,5], value=1)
|
109 |
+
nuc_size = st.sidebar.slider("Nucelus Size:\nLarger values lead to more diverse captions", min_value=0.0, max_value=1.0, value=0.8, step=0.05)
|
110 |
+
virtexModel.model.decoder.nucleus_size = nuc_size
|
|
|
111 |
|
112 |
image_file = sample_image
|
113 |
|