mipatov commited on
Commit
0fe7317
·
1 Parent(s): c3c88c9

t5 model name

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,7 +55,7 @@ def predict_t5(text, model, tokenizer, temperature):
55
  return generated_text
56
 
57
  gpt_model, gpt_tokenizer = get_model('mipatov/rugpt3_nb_descr', 'mipatov/rugpt3_nb_descr')
58
- t5_model, t5_tokenizer = get_model('mipatov/rugpt3_nb_descr', 'mipatov/rugpt3_nb_descr')
59
 
60
  # st.title("NeuroKorzh")
61
 
@@ -82,4 +82,4 @@ if button:
82
  st.text_area(label='', value=result, height=1000)
83
 
84
  except Exception:
85
- st.error("Ooooops, something went wrong. Please try again and report to me, tg: @vladyur")
 
55
  return generated_text
56
 
57
  gpt_model, gpt_tokenizer = get_model('mipatov/rugpt3_nb_descr', 'mipatov/rugpt3_nb_descr')
58
+ t5_model, t5_tokenizer = get_model('mipatov/rut5_nb_descr', 'mipatov/rut5_nb_descr')
59
 
60
  # st.title("NeuroKorzh")
61
 
 
82
  st.text_area(label='', value=result, height=1000)
83
 
84
  except Exception:
85
+ st.error("Ooooops, something went wrong. Please try again and report to me, tg: @mipatov")