import streamlit as st
from transformers import AutoTokenizer, AutoModel

with st.spinner('Loading tokenizer and model...'):
  tokenizer = AutoTokenizer.from_pretrained("malper/unikud")
  model = AutoModel.from_pretrained("malper/unikud")
st.success('Loaded!')

text = st.text_area('Enter Hebrew text')

if text:
  X = tokenizer([text], return_tensors='pt')
  logits = model(**X).logits.detach()[0]
  st.write(str(logits))

# x = st.slider('Select a value')
# st.write(x, 'squared is', x * x)