Spaces:
Runtime error
Runtime error
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
|