Mxode commited on
Commit
d712cdc
·
verified ·
1 Parent(s): 38ca9aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(model_options.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
 
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