File size: 402 Bytes
726e388 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import streamlit as st
from transformers import pipeline
@st.cache_resource # кэширование
def load_model():
return pipeline("sentiment-analysis") # скачивание модели
model = load_model()
query = st.text_input("Your query", value="I love Streamlit! 🎈")
if query:
result = model(query)[0] # классифицируем
st.write(query)
st.write(result) |