vladyur commited on
Commit
19d554b
·
1 Parent(s): 7d6f77f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_l
28
  return list(map(tokenizer.decode, out))[0]
29
 
30
 
31
- model, tokenizer = get_model('sberbank-ai/rugpt3medium_based_on_gpt2', 'SOME_CHECKPOINT.bin')
32
 
33
  st.title("NeuroKorzh")
34
  st.markdown("<img width=200px src='https://avatars.yandex.net/get-music-content/2399641/5d26d7e5.p.975699/m1000x1000'>",
 
28
  return list(map(tokenizer.decode, out))[0]
29
 
30
 
31
+ model, tokenizer = get_model('sberbank-ai/rugpt3medium_based_on_gpt2', 'korzh-medium_30epochs_1bs.bin')
32
 
33
  st.title("NeuroKorzh")
34
  st.markdown("<img width=200px src='https://avatars.yandex.net/get-music-content/2399641/5d26d7e5.p.975699/m1000x1000'>",