import streamlit as st
from utils import get_res

st.title('Tokenizers demo')

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


option = st.selectbox(
    'Choose a tokenizer',
    ['狗', '貓', '鸚鵡', '天竺鼠'])
'你的答案：', option


model_name = st.text_input('Model Name', 'deepseek-ai/deepseek-coder-1.3b-instruct')

'Your choice:', model_name

input_data = st.text_input('Input Sentence', 'Hello world!!!')


res = get_res(model_name=model_name, input_sentence=input_data, single_print=False)

st.write(res)