import streamlit as st
import torch
from transformers import BertTokenizer, BertForMaskedLM
from transformers import BertForSequenceClassification


tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
bert_classifier = BertForSequenceClassification.from_pretrained("bert_classifier", return_dict=True).train(False)

st.markdown("Hello!")
sentence = st.text_input("Please, enter your sentence. I will try to make it more positive :)")

encodings = tokenizer(sentence, return_tensors='pt', padding=True)
with torch.no_grad():
    logits = bert_classifier(**encodings).logits

st.markdown(logits)