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 |
|