test / app.py
Karpernik's picture
Create app.py
726e388 verified
raw
history blame contribute delete
402 Bytes
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)