patpizio commited on
Commit
fec2994
1 Parent(s): 42c9093

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,14 +43,14 @@ col1, col2 = st.columns(2)
43
  with col1:
44
  model_checkpoint = st.selectbox(
45
  "Model:",
46
- ("google/flan-t5-base", "google/flan-t5-xl")
47
  )
48
 
49
  with col2:
50
  temperature = st.slider('Temperature: ', min_value=0.0, max_value=1.0, value=0.5)
51
  top_p = st.slider('Top p: ', min_value=0.5, max_value=1.0, value=0.99)
52
  max_tokens = st.number_input('Max output length: ', min_value=1, max_value=64, format='%i')
53
- threshold = st.number_input('Min token score:: ', value=-np.inf)
54
 
55
 
56
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
43
  with col1:
44
  model_checkpoint = st.selectbox(
45
  "Model:",
46
+ ("google/flan-t5-base", "google/flan-t5-large", "google/flan-t5-xl")
47
  )
48
 
49
  with col2:
50
  temperature = st.slider('Temperature: ', min_value=0.0, max_value=1.0, value=0.5)
51
  top_p = st.slider('Top p: ', min_value=0.5, max_value=1.0, value=0.99)
52
  max_tokens = st.number_input('Max output length: ', min_value=1, max_value=64, format='%i')
53
+ threshold = st.number_input('Min token score:: ', value=-100.0)
54
 
55
 
56
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)