g8a9 commited on
Commit
da43319
1 Parent(s): 968ab83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -27,11 +27,11 @@ if gen_mode == "beam_search":
27
  "num_beams": num_beams,
28
  "early_stopping": early_stopping
29
  }
30
- else:
31
  do_sample = True
32
- top_k = st.sidebar.number_input("topk", value=30, min_value=0)
33
- top_p = st.sidebar.number_input("topk", value=0, min_value=0)
34
- temperature = st.sidebar.number_input("topk", value=0.7, min_value=0.0)
35
  gen_params = {
36
  "do_sample": do_sample,
37
  "top_k": top_k,
 
27
  "num_beams": num_beams,
28
  "early_stopping": early_stopping
29
  }
30
+ elif gen_mode == "sampling":
31
  do_sample = True
32
+ top_k = st.sidebar.number_input("top_k", value=30, min_value=0)
33
+ top_p = st.sidebar.number_input("top_p", value=0, min_value=0)
34
+ temperature = st.sidebar.number_input("temperature", value=0.7, min_value=0.0)
35
  gen_params = {
36
  "do_sample": do_sample,
37
  "top_k": top_k,