Spaces:
Sleeping
Sleeping
import streamlit as st | |
from funcs.nastya_funcs import predict_review, predict_bert | |
st.title('Restaurant reviews classifier') | |
text = st.text_input("Text to classify") | |
if text: | |
label, rating, time = predict_review(text) | |
col = st.columns(2) | |
col[0].write('Model: Tf-Idf + LogReg') | |
col[0].write(f"Отзыв: {rating}({label})") | |
col[0].write(f"Затраченное время: {time:.6f}с") | |
cls_name, name, time1 = predict_bert(text) | |
col[1].write('Model: Bert') | |
col[1].write(f"Отзыв: {name}({cls_name})") | |
col[1].write(f"Затраченное время: {time1:.6f}с") | |