Update app.py

app.py

@@ -5,14 +5,13 @@ from transformers import BertForSequenceClassification
 
 
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-bert_classifier = BertForSequenceClassification.from_pretrained(
-    "./bert_classifier", return_dict=True).train(False)
+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
+# with torch.no_grad():
+#     logits = bert_classifier(**encodings).logits
 
-st.markdown(logits)
+st.markdown(encodings)