Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def translate(text: str, model, tokenizer: PreTrainedTokenizerBase, **kwargs):
|
|
51 |
st.title("NanoTranslator-Demo")
|
52 |
|
53 |
st.sidebar.title("Options")
|
54 |
-
model_choice = st.sidebar.selectbox("Model", list(model_dict.keys()), index=list(
|
55 |
do_sample = st.sidebar.checkbox("do_sample", value=True)
|
56 |
max_new_tokens = st.sidebar.slider(
|
57 |
"max_new_tokens", min_value=1, max_value=256, value=64
|
|
|
51 |
st.title("NanoTranslator-Demo")
|
52 |
|
53 |
st.sidebar.title("Options")
|
54 |
+
model_choice = st.sidebar.selectbox("Model", list(model_dict.keys()), index=list(model_dict.keys()).index("NanoTranslator-XXL2"))
|
55 |
do_sample = st.sidebar.checkbox("do_sample", value=True)
|
56 |
max_new_tokens = st.sidebar.slider(
|
57 |
"max_new_tokens", min_value=1, max_value=256, value=64
|