Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=-
|
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)
|