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) |