Spaces:
Runtime error
Runtime error
gchhablani
commited on
Commit
•
053cee9
1
Parent(s):
ff1a6ad
Update app
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def read_markdown(path, parent="./sections/"):
|
|
42 |
|
43 |
|
44 |
checkpoints = ["./ckpt/ckpt-17499"] # TODO: Maybe add more checkpoints?
|
45 |
-
dummy_data = pd.read_csv("
|
46 |
|
47 |
st.set_page_config(
|
48 |
page_title="Multilingual Image Captioning",
|
@@ -57,9 +57,9 @@ st.write(
|
|
57 |
)
|
58 |
|
59 |
st.sidebar.title("Generation Parameters")
|
60 |
-
num_beams = st.sidebar.number_input(
|
61 |
-
temperature = st.sidebar.select_slider(
|
62 |
-
top_p = st.sidebar.select_slider(
|
63 |
|
64 |
|
65 |
image_col, intro_col = st.beta_columns([3, 8])
|
|
|
42 |
|
43 |
|
44 |
checkpoints = ["./ckpt/ckpt-17499"] # TODO: Maybe add more checkpoints?
|
45 |
+
dummy_data = pd.read_csv("references.tsv", sep="\t")
|
46 |
|
47 |
st.set_page_config(
|
48 |
page_title="Multilingual Image Captioning",
|
|
|
57 |
)
|
58 |
|
59 |
st.sidebar.title("Generation Parameters")
|
60 |
+
num_beams = st.sidebar.number_input("Number of Beams", min_value=2, max_value=10, value=4, step=1, help="Number of beams to be used in beam search.")
|
61 |
+
temperature = st.sidebar.select_slider("Temperature", options = list(np.arange(0.0,1.1, step=0.1)), value=1.0, help ="The value used to module the next token probabilities.", format_func=lambda x: f"{x:.2f}")
|
62 |
+
top_p = st.sidebar.select_slider("Top-P", options = list(np.arange(0.0,1.1, step=0.1)),value=1.0, help="Nucleus Sampling : If set to float < 1, only the most probable tokens with probabilities that add up to :obj:`top_p` or higher are kept for generation.", format_func=lambda x: f"{x:.2f}")
|
63 |
|
64 |
|
65 |
image_col, intro_col = st.beta_columns([3, 8])
|