Annorita's picture
add process function
35996ec
raw
history blame
545 Bytes
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)