Vladislawoo commited on
Commit
7901daa
·
1 Parent(s): c3bf9cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -88,18 +88,14 @@ def page_reviews_classification():
88
  st.write(f"Прогнозируемый класс: {prediction[0]}")
89
  st.write(f"Время вычисления: {elapsed_time:.2f} сек.")
90
 
91
- user_input_rnn = st.text_area("Введите текст отзыва для RNN модели:")
92
-
93
- if st.button("Классифицировать с RNN"):
94
  start_time = time.time()
95
  prediction_rnn = predict_text(user_input_rnn)
96
  elapsed_time = time.time() - start_time
97
  st.write(f"Прогнозируемый класс с RNN: {prediction_rnn}")
98
  st.write(f"Время вычисления: {elapsed_time:.2f} сек.")
99
 
100
- user_input_bert = st.text_area("Введите текст отзыва для BERT:")
101
-
102
- if st.button("Классифицировать (BERT)"):
103
  start_time = time.time()
104
  encoding = tokenizer.encode_plus(
105
  user_input_bert,
 
88
  st.write(f"Прогнозируемый класс: {prediction[0]}")
89
  st.write(f"Время вычисления: {elapsed_time:.2f} сек.")
90
 
91
+ elif st.button("Классифицировать с RNN"):
 
 
92
  start_time = time.time()
93
  prediction_rnn = predict_text(user_input_rnn)
94
  elapsed_time = time.time() - start_time
95
  st.write(f"Прогнозируемый класс с RNN: {prediction_rnn}")
96
  st.write(f"Время вычисления: {elapsed_time:.2f} сек.")
97
 
98
+ elif st.button("Классифицировать (BERT)"):
 
 
99
  start_time = time.time()
100
  encoding = tokenizer.encode_plus(
101
  user_input_bert,