gchhablani commited on
Commit
053cee9
1 Parent(s): ff1a6ad

Update app

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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("reference.tsv", sep="\t")
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(label="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(label="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(label = "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])
 
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])