vladyur commited on
Commit
763ddc9
·
1 Parent(s): bb72c45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -33,7 +33,7 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_l
33
  model, tokenizer = get_model('sberbank-ai/rugpt3medium_based_on_gpt2', 'korzh-medium_30epochs_1bs.bin')
34
 
35
  st.title("NeuroKorzh")
36
- st.markdown("<img width=200px src='https://avatars.yandex.net/get-music-content/2399641/5d26d7e5.p.975699/m1000x1000'>",
37
  unsafe_allow_html=True)
38
 
39
  st.markdown("\n")
@@ -45,6 +45,8 @@ if button:
45
  #try:
46
  result = predict(text, model, tokenizer)
47
  st.subheader('Max Korzh:')
48
- st.write(result)
 
 
49
  #except Exception:
50
  # st.error("Ooooops, something went wrong. Try again please and report to me, tg: @vladyur")
 
33
  model, tokenizer = get_model('sberbank-ai/rugpt3medium_based_on_gpt2', 'korzh-medium_30epochs_1bs.bin')
34
 
35
  st.title("NeuroKorzh")
36
+ st.markdown("<img width=400px src='https://avatars.yandex.net/get-music-content/2399641/5d26d7e5.p.975699/m1000x1000'>",
37
  unsafe_allow_html=True)
38
 
39
  st.markdown("\n")
 
45
  #try:
46
  result = predict(text, model, tokenizer)
47
  st.subheader('Max Korzh:')
48
+ lines = result.split('\n')
49
+ for line in lines:
50
+ st.write(line)
51
  #except Exception:
52
  # st.error("Ooooops, something went wrong. Try again please and report to me, tg: @vladyur")