Update app.py

app.py

@@ -1,6 +1,18 @@
-import torch
 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 :)")
-st.markdown(sentence)
+
+encodings = tokenizer(sentence, return_tensors='pt', padding=True)
+with torch.no_grad():
+    logits = bert_classifier(**encodings).logits
+
+st.markdown(logits)