import torch
import streamlit as st
import transformers

st.markdown("hello!")

model = transformers.AutoModel.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
tokenizer = transformers.AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")

user_input = st.text_input("Please enter something")

if len(user_input.split()) > 0:
    outputs = tokenizer(user_input, return_tensors="pt")
    st.markdown(f"{model(**outputs)}")