Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ no_repeat_ngram_size = st.sidebar.number_input("no repeat ngrams size", value=2,
|
|
20 |
num_return_sequences = st.sidebar.number_input("Generated sequences", value=3, min_value=1)
|
21 |
|
22 |
gen_mode = st.sidebar.selectbox("Generation mode", ["beam search", "sampling"])
|
23 |
-
if gen_mode == "
|
24 |
num_beams = st.sidebar.number_input("Beam size", value=5, min_value=1)
|
25 |
early_stopping = st.sidebar.checkbox("Early stopping", value=True)
|
26 |
gen_params = {
|
|
|
20 |
num_return_sequences = st.sidebar.number_input("Generated sequences", value=3, min_value=1)
|
21 |
|
22 |
gen_mode = st.sidebar.selectbox("Generation mode", ["beam search", "sampling"])
|
23 |
+
if gen_mode == "beam search":
|
24 |
num_beams = st.sidebar.number_input("Beam size", value=5, min_value=1)
|
25 |
early_stopping = st.sidebar.checkbox("Early stopping", value=True)
|
26 |
gen_params = {
|