Fix app

app.py

@@ -56,7 +56,7 @@ st.write(
 )
 
 st.sidebar.title("Generation Parameters")
-max_length = st.sidebar.number_input("Max Length", min_value=2, max_value=10, value=4, step=1, help="The maximum length of sequence to be generated.")
+max_length = st.sidebar.number_input("Max Length", min_value=16, max_value=128, value=64, step=1, help="The maximum length of sequence to be generated.")
 do_sample = st.sidebar.checkbox("Sample", value=False, help="Sample from the model instead of using beam search.")
 top_k = st.sidebar.number_input("Top K", min_value=10, max_value=200, value=50, step=1, help="The number of highest probability vocabulary tokens to keep for top-k-filtering.")
 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.")
@@ -127,7 +127,7 @@ with new_col1.beta_expander("Reference Caption"):
 sequence = ['']
 if new_col2.button("Generate Caption", help="Generate a caption in the specified language."):
     with st.spinner("Generating Sequence..."):
-        sequence = generate_sequence(transformed_image, num_beams, temperature, top_p)
+        sequence = generate_sequence(transformed_image, num_beams, temperature, top_p, do_sample, top_k, max_length)
 # print(sequence)
 
 if sequence!=['']: