Spaces:
Runtime error
Runtime error
| 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) |