vladyur commited on
Commit
e1458fb
·
1 Parent(s): ccd2dd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_l
32
  return list(map(tokenizer.decode, out))[0]
33
 
34
 
35
- model, tokenizer = get_model('sberbank-ai/rugpt3medium_based_on_gpt2', 'korzh-medium_30epochs_1bs.bin')
36
 
37
  # st.title("NeuroKorzh")
38
  # st.markdown("<img width=400px src='https://the-flow.ru/uploads/images/resize/830x0/adaptiveResize/05/06/06/42/25/8c7405840cd7.jpg'>",
 
32
  return list(map(tokenizer.decode, out))[0]
33
 
34
 
35
+ model, tokenizer = get_model('sberbank-ai/rugpt3medium_based_on_gpt2', 'korzh-medium_best_eval_loss.bin')
36
 
37
  # st.title("NeuroKorzh")
38
  # st.markdown("<img width=400px src='https://the-flow.ru/uploads/images/resize/830x0/adaptiveResize/05/06/06/42/25/8c7405840cd7.jpg'>",